gc collect on trainer teardown

Lightning-AI · Jul 29, 2021 · 6f3afb2 · 6f3afb2
1 parent ecf42db
commit 6f3afb2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Trainer to automate the training."""
+import gc
 import logging
 import os
 import traceback
@@ -982,6 +983,7 @@ def _post_dispatch(self):
         self.accelerator.teardown()
         self._active_loop.teardown()
         self.logger_connector.teardown()
+        gc.collect()
 
     def _dispatch(self):
         if self.evaluating: