down into cpp

pytorch · Apr 24, 2024 · 58359ca · 58359ca
1 parent e25e9d2
commit 58359ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_test_mnist.py b/test/test_test_mnist.py
@@ -91,8 +91,8 @@ def test():
   test_loss /= 20
   test_losses.append(test_loss)
   print('\nTest set: Avg. loss: {:.4f}, Accuracy: {}/{} ({:.0f}%)\n'.format(
-    test_loss, correct, len(test_loader.dataset),
-    100. * correct / len(test_loader.dataset)))
+    test_loss, correct, 20,
+    100. * correct / 20))
 
 # run test model
 test()