wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +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
tests/debug.py
+5 -5
View File
@@ -6,14 +6,14 @@ from pytorch_lightning.callbacks import ModelCheckpoint
import os
import shutil
import pytorch_lightning as ptl
import pytorch_lightning as pl
import torch
from torch.nn import functional as F
from torch.utils.data import DataLoader
from torchvision.datasets import MNIST
class CoolModel(ptl.LightningModule):
class CoolModel(pl.LightningModule):
def __init(self):
super(CoolModel, self).__init__()
@@ -43,15 +43,15 @@ class CoolModel(ptl.LightningModule):
def configure_optimizers(self):
return [torch.optim.Adam(self.parameters(), lr=0.02)]
@ptl.data_loader
@pl.data_loader
def tng_dataloader(self):
return DataLoader(MNIST('path/to/save', train=True), batch_size=32)
@ptl.data_loader
@pl.data_loader
def val_dataloader(self):
return DataLoader(MNIST('path/to/save', train=False), batch_size=32)
@ptl.data_loader
@pl.data_loader
def test_dataloader(self):
return DataLoader(MNIST('path/to/save', train=False), batch_size=32)