train_mnist.py 1.58 KB