debug

awaelchli · awaelchli · commit eec4ee2f77b5 · 2021-06-28T01:05:51.000+02:00
diff --git a/pytorch_lightning/loops/epoch/evaluation_epoch_loop.py b/pytorch_lightning/loops/epoch/evaluation_epoch_loop.py
@@ -114,6 +114,8 @@ def advance(
         # hook + store predictions
         self.on_evaluation_batch_end(output, batch, batch_idx, dataloader_idx)
 
+        del batch
+
         # log batch metrics
         self.trainer.logger_connector.update_eval_step_metrics()
 
diff --git a/pytorch_lightning/loops/epoch/training_epoch_loop.py b/pytorch_lightning/loops/epoch/training_epoch_loop.py
@@ -133,6 +133,9 @@ def advance(self, dataloader_iter: Iterator, **kwargs: Any) -> None:
         self.trainer.call_hook(
             'on_train_batch_end', processed_batch_end_outputs, batch, self.iteration_count, self._dataloader_idx
         )
+
+        del batch
+
         self.trainer.call_hook('on_batch_end')
         self.trainer.logger_connector.on_batch_end()
 

Original file line number	Diff line number	Diff line change
`@@ -133,6 +133,9 @@ def advance(self, dataloader_iter: Iterator, **kwargs: Any) -> None:`
`133`	`133`	`self.trainer.call_hook(`
`134`	`134`	`'on_train_batch_end', processed_batch_end_outputs, batch, self.iteration_count, self._dataloader_idx`
`135`	`135`	`)`
	`136`	`+`
	`137`	`+ del batch`
	`138`	`+`
`136`	`139`	`self.trainer.call_hook('on_batch_end')`
`137`	`140`	`self.trainer.logger_connector.on_batch_end()`
`138`	`141`