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

Merge pull request #52 from alok/ptl-pl

Browse Source 
Rename `ptl` to `pl`
This commit is contained in:
William Falcon
2019-08-07 09:09:15 -04:00
committed by GitHub
parent b4c14a4b77 8b9f021ee6
commit 549d0f66df
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
@@ -199,17 +199,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)