Merge pull request #122 from justusschock/fix-losstype-2dunet

Make returned losses and metrics scalar
delira-dev · Jun 5, 2019 · 62dccbe · 62dccbe
2 parents f7ae91b + 6954d17
commit 62dccbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/delira/models/segmentation/unet.py b/delira/models/segmentation/unet.py
@@ -239,13 +239,13 @@ def closure(model, data_dict: dict, optimizers: dict, losses={},
 
                     for key, crit_fn in losses.items():
                         _loss_val = crit_fn(preds["pred"], *data_dict.values())
-                        loss_vals[key] = _loss_val.detach()
+                        loss_vals[key] = _loss_val.item()
                         total_loss += _loss_val
 
                     with torch.no_grad():
                         for key, metric_fn in metrics.items():
                             metric_vals[key] = metric_fn(
-                                preds["pred"], *data_dict.values())
+                                preds["pred"], *data_dict.values()).item()
 
             if optimizers:
                 optimizers['default'].zero_grad()