From cd11b7de984eb9e4164799e4e8b644e3f2e99acf Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 2 Jul 2019 09:23:47 -0400
Subject: [PATCH] remove default tensor

---
 pytorch_lightning/root_module/root_module.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/pytorch_lightning/root_module/root_module.py b/pytorch_lightning/root_module/root_module.py
index 2349e7e4c0..969c3e647e 100644
--- a/pytorch_lightning/root_module/root_module.py
+++ b/pytorch_lightning/root_module/root_module.py
@@ -9,7 +9,6 @@ from pytorch_lightning.root_module.optimization import OptimizerConfig
 from pytorch_lightning.root_module.hooks import ModelHooks
 
 
-
 class LightningModule(GradInformation, ModelIO, OptimizerConfig, ModelHooks):
 
     def __init__(self, hparams):
@@ -40,10 +39,6 @@ class LightningModule(GradInformation, ModelIO, OptimizerConfig, ModelHooks):
         self._val_dataloader = None
         self._test_dataloader = None
 
-        if self.on_gpu:
-            print('running on gpu...')
-            torch.set_default_tensor_type(hparams.default_tensor_type)
-
     def forward(self, *args, **kwargs):
         """
         Expand model in into whatever you need.