fix 2333 (#2360)

2020-06-25 11:10:17 -04:00 · 2020-06-25 11:10:17 -04:00 · d22181714a
parent f2710bb500
commit d22181714a
1 changed files with 2 additions and 1 deletions
--- a/pytorch_lightning/trainer/trainer.py
+++ b/pytorch_lightning/trainer/trainer.py
@ -1193,7 +1193,8 @@ class Trainer(

        self.teardown('test')
        if self.is_function_implemented('teardown'):
-            self.model.teardown('test')
+            model_ref = self.get_model()
+            model_ref.teardown('test')

    def check_model_configuration(self, model: LightningModule):
        r"""