down into cpp

pytorch · Apr 24, 2024 · 2de7464 · 2de7464
1 parent bee343f
commit 2de7464
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_test_mnist.py b/test/test_test_mnist.py
@@ -77,7 +77,7 @@ def forward(self, x):
 train_losses = []
 train_counter = []
 test_losses = []
-test_counter = [i*len(train_loader.dataset) for i in range(n_epochs + 1)]
+test_counter = [i*20 for i in range(n_epochs + 1)]
 
 
 def test():