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

Rename ptl to pl

Browse Source 
Closes #46.
This commit is contained in:
Alok Singh
2019-08-06 23:02:55 -07:00
parent a79de1ec8e
commit 8b9f021ee6
4 changed files with 29 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/lightning_module_template.py
+4 -4
View File
@@ -13,7 +13,7 @@ from torch import optim
from torch.utils.data import DataLoader
from torch.utils.data.distributed import DistributedSampler
import pytorch_lightning as ptl
import pytorch_lightning as pl
from pytorch_lightning.root_module.root_module import LightningModule
@@ -195,17 +195,17 @@ class LightningTemplateModel(LightningModule):
return loader
@ptl.data_loader
@pl.data_loader
def tng_dataloader(self):
print('tng data loader called')
return self.__dataloader(train=True)
@ptl.data_loader
@pl.data_loader
def val_dataloader(self):
print('val data loader called')
return self.__dataloader(train=False)
@ptl.data_loader
@pl.data_loader
def test_dataloader(self):
print('test data loader called')
return self.__dataloader(train=False)