diff --git a/README.md b/README.md index 74f3f61fc1..5d61102c54 100644 --- 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() ```