diff --git a/beginner_source/basics/optimization_tutorial.py b/beginner_source/basics/optimization_tutorial.py index be17289fd19..e10496fabb6 100644 --- a/beginner_source/basics/optimization_tutorial.py +++ b/beginner_source/basics/optimization_tutorial.py @@ -160,7 +160,7 @@ def train_loop(dataloader, model, loss_fn, optimizer): optimizer.step() if batch % 100 == 0: - loss, current = loss.item(), batch * len(X) + loss, current = loss.item(), (batch + 1) * len(X) print(f"loss: {loss:>7f} [{current:>5d}/{size:>5d}]") diff --git a/beginner_source/basics/quickstart_tutorial.py b/beginner_source/basics/quickstart_tutorial.py index 8cba94bcc04..366a4193d0b 100644 --- a/beginner_source/basics/quickstart_tutorial.py +++ b/beginner_source/basics/quickstart_tutorial.py @@ -151,7 +151,7 @@ def train(dataloader, model, loss_fn, optimizer): optimizer.step() if batch % 100 == 0: - loss, current = loss.item(), batch * len(X) + loss, current = loss.item(), (batch + 1) * len(X) print(f"loss: {loss:>7f} [{current:>5d}/{size:>5d}]") ##############################################################################