fix typo/bug in validation logging

constantinpape · Jul 12, 2022 · 5fdbc1a · 5fdbc1a
1 parent aa8a1ea
commit 5fdbc1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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