Skip to content

Commit 7553fbe

Browse files
authored
Fix validation loss logging (open-mmlab#1494)
1 parent e231583 commit 7553fbe

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

mmseg/models/segmentors/base.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,9 +155,14 @@ def val_step(self, data_batch, optimizer=None, **kwargs):
155155
losses = self(**data_batch)
156156
loss, log_vars = self._parse_losses(losses)
157157

158+
log_vars_ = dict()
159+
for loss_name, loss_value in log_vars.items():
160+
k = loss_name + '_val'
161+
log_vars_[k] = loss_value
162+
158163
outputs = dict(
159164
loss=loss,
160-
log_vars=log_vars,
165+
log_vars=log_vars_,
161166
num_samples=len(data_batch['img_metas']))
162167

163168
return outputs

0 commit comments

Comments
 (0)