fix: increasing training time over epochs

Alexandre-Delplanque · Nov 30, 2023 · 39e317a · 39e317a
1 parent 4baeab1
commit 39e317a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/animaloc/train/trainers.py b/animaloc/train/trainers.py
@@ -524,7 +524,7 @@ def _train(
                 wandb.log(loss_dict)
 
             self.losses = sum(loss for loss in loss_dict.values())
-            batches_losses.append(self.losses)
+            batches_losses.append(self.losses.detach())
 
             loss_dict_reduced = reduce_dict(loss_dict)
             losses_reduced = sum(loss for loss in loss_dict_reduced.values())