train_mnist_cnn.py #coding: utf-8 import numpy as np import chainer from chainer import cuda import chainer.functions as F from chainer import optimizers import time from sklearn.datasets import fetch_mldata from sklearn.cross_validation import train_test_split import pylab import matplotlib.pyplot as plt gpu_flag = 0 if gpu_flag >= 0: cuda.check_cuda_available() xp = cuda.cupy if gpu_flag >= 0 el