From 301a4992f45a63b9fbcecfbf04626e3b68aa4956 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 26 Jun 2019 18:24:47 -0400
Subject: [PATCH] removed self.model refs

---
 pytorch_lightning/models/trainer.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 241b2401..434e5060 100644
--- 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