diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 6a5673052521..283c9d59a549 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -227,7 +227,7 @@ def on_fit_epoch_end(self, vals, epoch, best_fitness, fi): x = dict(zip(self.keys, vals)) print( - f"Epoch: {epoch}, Training Loss: {vals[0].cpu().numpy().tolist()}, Validation Loss: {vals[7]}" + f"Epoch: {epoch}, Training Loss: {vals[0].cpu().numpy().tolist()}, Validation Loss: {vals[7]}, Validation mAP_0.5: {vals[5]}, Validation mAP_0.5:0.95: {vals[6]}" ) if self.csv: