train_mnist.py 4.19 KB