/S7-APP1

Primary LanguagePython

run test

python -m unittest discover . '*test.py'

train network

python train_mnist.py --learning_rate=0.01 --batch_size=20 --epoch_count=50 --output_path=./checkpoint_v1

evaluate a pretrained network

python train_mnist.py --learning_rate=0.01 --batch_size=20 --epoch_count=0 --output_path=./checkpoint_v1 --checkpoint_path=./checkpoint_v1/checkpoint_epoch_37.pkl