diff --git a/train.py b/train.py
index 0efa836..2ad094d 100644
--- a/train.py
+++ b/train.py
@@ -13,7 +13,7 @@ BETA1 = 0.9
 BETA2 = 0.999
 EPSILON = 1e-8
 EPOCHS = 10
-BATCH_SIZE = 60
+BATCH_SIZE = 800
 HIDDEN_LAYERS = 4
 
 # Training images. It's a numpy array of size 60000x784.