fix outputs issue

Lightning-AI · Mar 12, 2021 · bce1a2c · bce1a2c
1 parent 174f767
commit bce1a2c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py
@@ -204,9 +204,6 @@ def __run_eval_epoch_end(self, num_dataloaders):
         # with a single dataloader don't pass an array
         outputs = self.outputs
 
-        # free memory
-        self.outputs = []
-
         eval_results = outputs
         if num_dataloaders == 1:
             eval_results = outputs[0]
@@ -321,6 +318,10 @@ def on_evaluation_epoch_end(self, *args, **kwargs):
 
     def call_on_evaluation_epoch_end_hook(self):
         outputs = self.outputs
+
+        # free memory
+        self.outputs = []
+
         model_ref = self.trainer.lightning_module
         hook_name = "on_test_epoch_end" if self.trainer.testing else "on_validation_epoch_end"