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

@ -41,7 +41,6 @@ optimizer = torch.optim.SGD(net.parameters(), lr=0.5)
loss_func = torch.nn.MSELoss() # this is for regression mean squared loss
plt.ion() # something about plotting
plt.show()
for t in range(100):
prediction = net(x) # input x and predict based on x