目前可用的mnist_dataset下载

[in]
【注】根据需要改变one_hot参数的值

import tensorflow as tf

# Import MINST data
tf.logging.set_verbosity(tf.logging.ERROR)#不显示错误信息
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("/tmp/data/", one_hot=False,
                                  source_url='http://yann.lecun.com/exdb/mnist/')

[out]

对于模型训练时常提示MNIST_DATA数据格式不对，此时需要将数据转换为onehot编码。

# Start training
with tf.Session() as sess: