mnist_code_examples Installation python 3.6.1 tensorflow 2.1.0 scikit-learn 0.22.1 Model training Model checkpoints are stored in saved_models directory. python train.py Define custom epoch number and batch size python train.py --epoch 50 --batch_size 100