TensorFlow实战google深度学习框架 代码报错
黄色TensorFlow实战第五章
5.5中的代码,在使用python3.5的情况下会报错(可能高版本都会有问题)
主要问题:
mnist_train.py中
1.书中的cross_entropy部分,需要加上logits=, labels=
cross_entropy = tf.nn.sparse_softmax_cross_entropy_with_logits( logits=y, labels=tf.argmax(y_, 1) )
2.saver函数存在问题
可以选择直接注释掉,也可以修改一下路劲
MODEL_SAVE_PATH = "\path\\to\model" MODEL_NAME = "model.ckpt"
3.initialize_all_variables报错
这个方法已经不能用了,需要用如下代码替换:
with tf.Session() as sess: # tf.initialize_all_variables.run() tf.global_variables_initializer().run()