[Fix]: Fix logging of validation loss (#9663)

open-mmlab · Jan 28, 2023 · fd76d2b · fd76d2b
1 parent 08f77b5
commit fd76d2b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mmdet/models/detectors/base.py b/mmdet/models/detectors/base.py
@@ -263,8 +263,13 @@ def val_step(self, data, optimizer=None):
         losses = self(**data)
         loss, log_vars = self._parse_losses(losses)
 
+        log_vars_ = dict()
+        for loss_name, loss_value in log_vars.items():
+            k = loss_name + '_val'
+            log_vars_[k] = loss_value
+
         outputs = dict(
-            loss=loss, log_vars=log_vars, num_samples=len(data['img_metas']))
+            loss=loss, log_vars=log_vars_, num_samples=len(data['img_metas']))
 
         return outputs