add train_mnist datasets