Merge pull request #150 from justusschock/trainer_fixes

add metric conversion
delira-dev · Jun 17, 2019 · 8da5c32 · 8da5c32
2 parents 9cd0c1c + 0850487
commit 8da5c32
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/delira/training/base_trainer.py b/delira/training/base_trainer.py
@@ -445,6 +445,8 @@ def reduce_fn(batch):
 
                 total_metrics.update(val_metrics)
 
+            _, total_metrics = self._convert_to_npy_fn(**total_metrics)
+
             for k, v in total_metrics.items():
                 total_metrics[k] = reduce_fn(v)