From b563cfe598643eba11f57c04da525a2b251245af Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 11:48:28 -0400 Subject: [PATCH] testing slurm ddp --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index f18ebb11..d7062257 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -271,7 +271,7 @@ class Trainer(TrainerIO): # ----------------------------- def fit(self, model): # set local properties on the model - self.model.on_gpu = self.on_gpu + model.on_gpu = self.on_gpu # transfer data loaders from model self.__get_dataloaders(model)