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

Deployed b914866 with MkDocs version: 1.0.4

Browse Source
This commit is contained in:
William Falcon
2019-07-25 11:12:53 -05:00
parent 4c60f2e8c3
commit a19dd86e4e
4 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
LightningModule/RequiredTrainerInterface/index.html
+3 -2
View File
@@ -925,8 +925,9 @@ from torchvision.datasets import MNIST
class CoolModel(ptl.LightningModule):
def __init(self):
super(CoolModel, self).__init__()
# not the best model...
self.l1 = torch.nn.Linear(28*28, 10)
self.l1 = torch.nn.Linear(28 * 28, 10)
def forward(self, x):
return torch.relu(self.l1(x))
@@ -961,7 +962,7 @@ class CoolModel(ptl.LightningModule):
@ptl.data_loader
def test_dataloader(self):
return DataLoader(MNIST('path/to/save', train=False), batch_size=32)
return DataLoader(MNIST('path/to/save', train=False), batch_size=32)
</code></pre>
<hr />