Skip to content

Commit fd76d2b

Browse files
authored
[Fix]: Fix logging of validation loss (#9663)
1 parent 08f77b5 commit fd76d2b

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

mmdet/models/detectors/base.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,8 +263,13 @@ def val_step(self, data, optimizer=None):
263263
losses = self(**data)
264264
loss, log_vars = self._parse_losses(losses)
265265

266+
log_vars_ = dict()
267+
for loss_name, loss_value in log_vars.items():
268+
k = loss_name + '_val'
269+
log_vars_[k] = loss_value
270+
266271
outputs = dict(
267-
loss=loss, log_vars=log_vars, num_samples=len(data['img_metas']))
272+
loss=loss, log_vars=log_vars_, num_samples=len(data['img_metas']))
268273

269274
return outputs
270275

0 commit comments

Comments
 (0)