fix path problem in train_mnist.sh