andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug for windows

Browse Source
This commit is contained in:
Morvan Zhou
2018-05-04 11:39:52 +10:00
parent 46c516fc10
commit 68396f5972
1 changed files with 36 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tutorial-contents/306_optimizer.py
View File

@ -43,6 +43,7 @@ class Net(torch.nn.Module):
x = self.predict(x) # linear output x = self.predict(x) # linear output
return x return x
if __name__ == '__main__':
# different nets # different nets
net_SGD = Net() net_SGD = Net()
net_Momentum = Net() net_Momentum = Net()