mnist test case

pytorch · Feb 28, 2024 · 03f9a62 · 03f9a62
1 parent e493f69
commit 03f9a62
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/test_train_mp_mnist.py b/test/test_train_mp_mnist.py
@@ -171,8 +171,16 @@ def test_loop_fn(loader):
     correct = 0
     model.eval()
     interator_local = 0
+
     print("loader: ", loader)
     print("type loader: ", type(loader))
+
+    loop_length = loader.__len__
+    test_dataset = enumerate(loader)
+
+    print("test_dataset: ", test_dataset)
+    print("type test_dataset: ", type(test_dataset))
+
     for data, target in loader:
       interator_local = interator_local + 1
       print("test_loop_fn for loop: ", interator_local)