need zero grad now

2020-12-07 23:10:43 -08:00 · 2020-12-07 23:10:43 -08:00 · c63f950348
parent b355cd2571
commit c63f950348
1 changed files with 1 additions and 0 deletions
--- a/README.md
+++ b/README.md
@ -73,6 +73,7 @@ optim = optim.SGD([model.l1, model.l2], lr=0.001)

 out = model.forward(x)
 loss = out.mul(y).mean()
+optim.zero_grad()
 loss.backward()
 optim.step()
 ```