A flexible framework of neural networks for deep learning - chainer/train_mnist.py at v4.0.0 · chainer/chainer