lorenzo/mnist
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

increasing batch size makes the training process sensibly faster.

Browse source
This commit is contained in:
Lorenzo Torres 2025-11-14 12:13:02 +01:00
parent cb34f2c5e9
commit cc861f6ffb
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
train.py
View file

@ -13,7 +13,7 @@ BETA1 = 0.9
BETA2 = 0.999 BETA2 = 0.999
EPSILON = 1e-8 EPSILON = 1e-8
EPOCHS = 10 EPOCHS = 10
BATCH_SIZE = 60 BATCH_SIZE = 800
HIDDEN_LAYERS = 4 HIDDEN_LAYERS = 4
# Training images. It's a numpy array of size 60000x784. # Training images. It's a numpy array of size 60000x784.