need zero grad now

2026-01-09 15:08:02 -05:00 · 2020-12-07 23:10:43 -08:00
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()
 ```