Fix validation loss logging (#1494)

open-mmlab · Apr 22, 2022 · 7553fbe · 7553fbe
1 parent e231583
commit 7553fbe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py
@@ -155,9 +155,14 @@ def val_step(self, data_batch, optimizer=None, **kwargs):
         losses = self(**data_batch)
         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,
+            log_vars=log_vars_,
             num_samples=len(data_batch['img_metas']))
 
         return outputs