diff --git a/torch_em/trainer/default_trainer.py b/torch_em/trainer/default_trainer.py index fec07fbc..921bfd68 100644 --- a/torch_em/trainer/default_trainer.py +++ b/torch_em/trainer/default_trainer.py @@ -544,5 +544,5 @@ def _validate_impl(self, forward_context): metric_val /= len(self.val_loader) loss_val /= len(self.val_loader) if self.logger is not None: - self.logger.log_validation(self._iteration, metric, loss, x, y, pred) + self.logger.log_validation(self._iteration, metric_val, loss_val, x, y, pred) return metric_val