kevin

Author: Faten848

examples/mnist/FFsnn_mnist

@@ -41,8 +41,8 @@ def main():
 
     print_interval = 10 # Print training stats every 10 batches (adjust as needed for large batch sizes)
 
-    max_train_samples = 128    # Only 128 samples = 4 batches
-    max_test_samples = 64      # Only 64 test samples
+    max_train_samples = 2000    # Only 128 samples = 4 batches
+    max_test_samples = 500      # Only 64 test samples
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     print(f"Using device: {device}")