Closed kogolobo closed 8 months ago
Quick overview:
main.py
model.py
data.py
trainer.py
Example run: python main.py --epochs 10 --lr 0.01 --train_batch_size 32 --eval_batch_size 32
python main.py --epochs 10 --lr 0.01 --train_batch_size 32 --eval_batch_size 32
Quick overview:
main.py-- main entry point in the codemodel.py-- defines simple feedforward netdata.py-- creates a dataset abstraction, implements MNIST and CIFAR10 datasetstrainer.py-- encapsulates training/evaluation loopsExample run:
python main.py --epochs 10 --lr 0.01 --train_batch_size 32 --eval_batch_size 32