fix bug for windows

2018-05-04 11:39:52 +10:00
parent 46c516fc10
commit 68396f5972
1 changed files with 36 additions and 35 deletions
--- a/tutorial-contents/306_optimizer.py
+++ b/tutorial-contents/306_optimizer.py
@ -43,6 +43,7 @@ class Net(torch.nn.Module):
        x = self.predict(x)             # linear output
        return x

+if __name__ == '__main__':
    # different nets
    net_SGD         = Net()
    net_Momentum    = Net()