diff --git a/beginner_source/basics/quickstart_tutorial.py b/beginner_source/basics/quickstart_tutorial.py index 07a1be517d1..2bb4622d4e4 100644 --- a/beginner_source/basics/quickstart_tutorial.py +++ b/beginner_source/basics/quickstart_tutorial.py @@ -152,9 +152,9 @@ def train(dataloader, model, loss_fn, optimizer): loss = loss_fn(pred, y) # Backpropagation + optimizer.zero_grad() loss.backward() optimizer.step() - optimizer.zero_grad() if batch % 100 == 0: loss, current = loss.item(), (batch + 1) * len(X)