fix 2333 (#2360)

Lightning-AI · Jun 25, 2020 · d221817 · d221817
1 parent f2710bb
commit d221817
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1193,7 +1193,8 @@ def test(
 
         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"""