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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值