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

@ -58,7 +58,6 @@ optimizer_drop = torch.optim.Adam(net_dropped.parameters(), lr=0.01)
loss_func = torch.nn.MSELoss()
plt.ion() # something about plotting
plt.show()
for t in range(500):
pred_ofit = net_overfitting(x)