Merge pull request #73 from JonasHell/main

fix typo/bug in validation logging
constantinpape · Jul 12, 2022 · 651a830 · 651a830
2 parents aa8a1ea + 5fdbc1a
commit 651a830
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