小编给大家分享一下在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文件生效。
以上是“在TensorFlow中如何屏蔽warning”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。