Update mnist_neural_net_shallow.yml