From 347a56173d9fb9da60ec8e95c8ffcd27bcf236ac Mon Sep 17 00:00:00 2001
From: brucekimrokcmu <kwangkyk@alumni.cmu.edu>
Date: Sat, 14 Oct 2023 09:49:32 -0700
Subject: [PATCH] add drop_last in mnist eager example

---
 examples/eager_mlp/mlp_eager_simple.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/eager_mlp/mlp_eager_simple.py b/examples/eager_mlp/mlp_eager_simple.py
index 947d0da6e..49a37ba92 100644
--- a/examples/eager_mlp/mlp_eager_simple.py
+++ b/examples/eager_mlp/mlp_eager_simple.py
@@ -39,7 +39,7 @@ def get_train_loader(self):
 
     def get_test_loader(self):
         return DataLoader(
-            dataset=self.mnist_testset, batch_size=self.batch_size, shuffle=False
+            dataset=self.mnist_testset, batch_size=self.batch_size, shuffle=False, drop_last=True,
         )
 
 
@@ -70,7 +70,7 @@ def infer_iteration(model, images):
 def infer():
     # Example Parameters
     config = {
-        "batch_size": 100,
+        "batch_size": 64,
         "learning_rate": 0.001,
         "num_epochs": 10,
     }