onnxruntime/samples/python/mnist
..
mnist_training.py