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()

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页