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

updated args

Browse Source
This commit is contained in:
William Falcon
2019-06-25 19:35:11 -04:00
parent e801914d1d
commit c54dd94295
3 changed files with 16 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/examples/example_model.py
+3 -2
View File
@@ -6,7 +6,7 @@ from torchvision.datasets import MNIST
import torchvision.transforms as transforms
import torch
import torch.nn.functional as F
import os
import os, pdb
class ExampleModel(RootModule):
@@ -40,8 +40,9 @@ class ExampleModel(RootModule):
# ---------------------
# TRAINING
# ---------------------
def forward(self, x):
def forward(self, x, a):
pdb.set_trace()
x = self.c_d1(x)
x = F.tanh(x)
x = self.c_d1_bn(x)