From 7553fbe948dd258400cd574feb66a1d6e1316d8a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mladen=20Ferne=C5=BEir?=
 <mfernezir@users.noreply.github.com>
Date: Fri, 22 Apr 2022 09:25:32 +0200
Subject: [PATCH] Fix validation loss logging (#1494)

---
 mmseg/models/segmentors/base.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py
index 9b22a7ca9..76dc8f075 100644
--- a/mmseg/models/segmentors/base.py
+++ b/mmseg/models/segmentors/base.py
@@ -155,9 +155,14 @@ class BaseSegmentor(BaseModule, metaclass=ABCMeta):
         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