-
Dev by SCIMTA EX-team:
-
Thanks to:
-
Reference:
# Clone source code
git clone https://github.com/trongthanht3/neural-network-from-scatch
cd neural-network-from-scatch
python main.py --help
usage: main.py [-h] [--optimizer OPTIMIZER] [--batch_size BATCH_SIZE]
[--epochs EPOCHS] [--save_model SAVE_MODEL]
optional arguments:
-h, --help show this help message and exit
--optimizer OPTIMIZER
choose optimzer to train
--batch_size BATCH_SIZE
batch size
--epochs EPOCHS nums of epoch
--save_model SAVE_MODEL
True/False save model
# Example
python main.py --optimizer=sgd --batch_size=128 --epochs=25 --save_model=False