agentic-fix(try-2): test_accurac issue

Author: raselmandol

train_mnist.py

@@ -126,8 +126,8 @@ def forward(self, x):
         inputs, labels = inputs.to(device), labels.to(device)
         outputs = model(inputs)
         test_loss += criterion(outputs, labels).item()
-        batch_correct = (outputs.argmax(1) == labels).sum().item()
-        correct += batch_correct
+    batch_correct = (outputs.argmax(1) == labels).sum().item()
+    correct += batch_correct
         batch_acc = batch_correct / len(inputs)
         test_accuracies.append(batch_acc)