From 802d68671f544c32347a2f1e780b80fcc2d8c9e7 Mon Sep 17 00:00:00 2001
From: anttikup <113663027+anttikup@users.noreply.github.com>
Date: Tue, 23 Jan 2024 11:29:54 +0200
Subject: [PATCH] Fix calculation of current item number in
 optimization_tutorial.py

Current item calculation gave the wrong number when dataset doesn't split to batches equally.
---
 beginner_source/basics/optimization_tutorial.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/beginner_source/basics/optimization_tutorial.py b/beginner_source/basics/optimization_tutorial.py
index 93aed46161d..c6c327f8511 100644
--- a/beginner_source/basics/optimization_tutorial.py
+++ b/beginner_source/basics/optimization_tutorial.py
@@ -163,7 +163,7 @@ def train_loop(dataloader, model, loss_fn, optimizer):
         optimizer.zero_grad()
 
         if batch % 100 == 0:
-            loss, current = loss.item(), (batch + 1) * len(X)
+            loss, current = loss.item(), batch * batch_size + len(X)
             print(f"loss: {loss:>7f}  [{current:>5d}/{size:>5d}]")