__call__ to forward

soumith · soumith · commit 12e213c52994 · 2016-12-01T04:35:53.000-08:00
diff --git a/mnist/main.py b/mnist/main.py
@@ -44,7 +44,7 @@ def __init__(self):
             softmax = nn.LogSoftmax(),
         )
 
-    def __call__(self, x):
+    def forward(self, x):
         x = self.relu(self.pool1(self.conv1(x)))
         x = self.relu(self.pool2(self.conv2(x)))
         x = x.view(-1, 800)

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ def __init__(self):`
`44`	`44`	`softmax = nn.LogSoftmax(),`
`45`	`45`	`)`
`46`	`46`
`47`		`- def __call__(self, x):`
	`47`	`+ def forward(self, x):`
`48`	`48`	`x = self.relu(self.pool1(self.conv1(x)))`
`49`	`49`	`x = self.relu(self.pool2(self.conv2(x)))`
`50`	`50`	`x = x.view(-1, 800)`