From e493f69a114341ea524dbe02d54caa077c9d736a Mon Sep 17 00:00:00 2001
From: manfeibaigithub <manfei@google.com>
Date: Wed, 28 Feb 2024 23:20:04 +0000
Subject: [PATCH] mnist test case

---
 test/test_train_mp_mnist.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/test/test_train_mp_mnist.py b/test/test_train_mp_mnist.py
index 3b078d22fab..fab43ebaaa9 100644
--- a/test/test_train_mp_mnist.py
+++ b/test/test_train_mp_mnist.py
@@ -170,7 +170,12 @@ def test_loop_fn(loader):
     total_samples = 0
     correct = 0
     model.eval()
+    interator_local = 0
+    print("loader: ", loader)
+    print("type loader: ", type(loader))
     for data, target in loader:
+      interator_local = interator_local + 1
+      print("test_loop_fn for loop: ", interator_local)
       output = model(data)
       pred = output.max(1, keepdim=True)[1]
       correct += pred.eq(target.view_as(pred)).sum()