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

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
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)