wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove default tensor

Browse Source
This commit is contained in:
William Falcon
2019-07-02 09:23:47 -04:00
parent 2eba85d02c
commit cd11b7de98
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/root_module.py
-5
View File
@@ -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.