TensorFlow中屏蔽warning的方法

    TensorFlow的日志级别分为以下三种:
TF_CPP_MIN_LOG_LEVEL = 1 //默认设置,为显示所有信息
TF_CPP_MIN_LOG_LEVEL = 2 //只显示error和warining信息
TF_CPP_MIN_LOG_LEVEL = 3 //只显示error信息
所以,当TensorFlow出现警告信息,又不想让警告信息显示时,可进行如下设置:
一、python环境下
通过在python文件中添加如下两行代码,设置TensorFlow日志输出级别

impot os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

二、linux c/c++调用环境下
在linux环境下,我们可以通过linux命令,屏蔽掉TensorFlow的warning:

1、暂时屏蔽warning:
在linux终端中输入命令:export TF_CPP_MIN_LOG_LEVEL=2,当该终端关闭,既失效;
2、永久屏蔽warning:

修改/etc/profile文件,在该文件中加入 export TF_CPP_MIN_LOG_LEVEL=2,但是修改后需要重启
系统或者使用source  /etc/profile命令,使修改后的profile文件生效。
  • 9
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 11
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值