mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Fix validation loss logging (#1494)

Browse Source
This commit is contained in:
Mladen Fernežir 2022-04-22 09:25:32 +02:00 committed by GitHub
parent aa68bf15b7
commit 9e17a0d364
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmseg/models/segmentors/base.py
View File

@ -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