train_mnist.py 1.4 KB