This commit is contained in:
Morvan Zhou
2017-05-23 10:37:51 +10:00
committed by Morvan Zhou
parent 30c0070497
commit d87529ed89
10 changed files with 17 additions and 10 deletions

View File

@ -47,7 +47,6 @@ optimizer = torch.optim.SGD(net.parameters(), lr=0.02)
loss_func = torch.nn.CrossEntropyLoss() # the target label is NOT an one-hotted
plt.ion() # something about plotting
plt.show()
for t in range(100):
out = net(x) # input x and predict based on x