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.