TensorFlow实战中的经验与debug记录(持续更新)

1 TensorFlow版本问题

# 报错信息
early_stopping_hook = tf.contrib.estimator.stop_if_no_decrease_hook(
AttributeError: module 'tensorflow.contrib.estimator' has no attribute 'stop_if_no_decrease_hook'

查看了本机的 tf版本发现是1.14,程序源码的版本是1.11 。查找了tensorflow的文档,发现 stop_if_no_decrease_hook 这个函数换位置了,不在这个 tensorflow.contrib.estimator 里面
在1.14版本中 应该是

tf.estimator.experimental.stop_if_no_decrease_hook

2 安装tensorflow-gpu版本

在GPU上训练模型比CPU快太多倍了,所以一定安装tensorflow-gpu版本!

# 建议用conda安装,这样不用手动安装对应版本的cudnn和cuda
conda install tensorflow-gpu==1.14.0
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值