From 22f4d6e26e7d54b79eb06647ac666dc26914df61 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 18 Jul 2019 11:49:28 -0400
Subject: [PATCH] set dp as default backend

---
 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 36dfcf76..f30f3173 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -366,6 +366,7 @@ class Trainer(TrainerIO):
         # filter out the weights that were done on gpu so we can load on good old cpus
         self.optimizers = model.configure_optimizers()
 
+        model.cuda()
         model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids)
 
         # run through amp wrapper