onnxruntime/orttraining/pytorch_frontend_examples
..
mnist_training.py