From ab1e5dd7c87f79d4b4a014d74511971eb202788a Mon Sep 17 00:00:00 2001 From: ananthsub Date: Wed, 28 Oct 2020 12:40:00 -0700 Subject: [PATCH] Update accelerator.py --- pytorch_lightning/accelerators/accelerator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/accelerators/accelerator.py b/pytorch_lightning/accelerators/accelerator.py index 8e1969cc9368e..1e0e9145d4360 100644 --- a/pytorch_lightning/accelerators/accelerator.py +++ b/pytorch_lightning/accelerators/accelerator.py @@ -93,7 +93,7 @@ def backward(self, closure_loss, optimizer, opt_idx, *args, **kwargs): ) else: # do backward pass - model = self.trainer.get_model() + model = self.trainer.model model.backward(closure_loss, optimizer, opt_idx, *args, **kwargs) # once backward has been applied, release graph