release v0.122

Lightning-AI · Jul 2, 2019 · 0262f23 · 0262f23
1 parent 8fde5e4
commit 0262f23
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
@@ -280,6 +280,7 @@ def fit(self, model):
         # put on gpu if needed
         if self.on_gpu:
             model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids)
+            model.cuda(self.data_parallel_device_ids[0])
 
         # run tiny validation to make sure program won't crash during val
         _ = self.validate(model, self.val_dataloader, max_batches=self.nb_sanity_val_steps)

diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 # http://blog.ionelmc.ro/2014/05/25/python-packaging/
 setup(
     name="pytorch-lightning",
-    version='0.121',
+    version='0.122',
     description="The Keras for ML researchers using PyTorch",
     author="William Falcon",
     author_email="waf2107@columbia.edu",
@@ -19,7 +19,7 @@
     install_requires=[
         "torch>=1.1.0",
         "tqdm",
-        "test-tube>=0.651",
+        "test-tube>=0.652",
         "tensorflow>=1.14.0"
     ],
     packages=find_packages(),