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

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

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