From d52c92e09d30faef6ef5d33536966b0e79df338a Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 3 Jul 2019 15:26:19 -0400
Subject: [PATCH] added single node distdataparallel

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

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index eedf955b..3a182e1d 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -284,7 +284,7 @@ class Trainer(TrainerIO):
         # when GPU is called, spawn off a single worker for each gpu
         if self.on_gpu:
             rank = 0
-            # self.model = model
+            self.model = model
             mp.spawn(dummy, nprocs=len(self.data_parallel_device_ids), args=(self, ))
         else:
             self.__run_pretrain_routine(model)
@@ -292,7 +292,6 @@ class Trainer(TrainerIO):
     def __getstate__(self):
         """ This is called before pickling. """
         state = self.__dict__.copy()
-        # del state['cluster']
         del state['experiment']
         return state