edit load_mnist