removed self.model refs

2019-06-26 18:24:47 -04:00 · 2019-06-26 18:24:47 -04:00 · 301a4992f4
parent 42fe76f794
commit 301a4992f4
1 changed files with 0 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -268,7 +268,6 @@ class Trainer(TrainerIO):

        # put on gpu if needed
        if self.on_gpu:
-            model.cuda()
            model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids)

        # run tiny validation to make sure program won't crash during val