From cc861f6ffb4d0c115e58e9be3dca4f73bc32caed Mon Sep 17 00:00:00 2001 From: Lorenzo Torres Date: Fri, 14 Nov 2025 12:13:02 +0100 Subject: [PATCH] increasing batch size makes the training process sensibly faster. --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.