欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

在TensorFlow中屏蔽warning的方式

时间:2022-05-06 09:21:31|栏目:Python代码|点击:

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文件生效。

上一篇:使用Python实现给企业微信发送消息功能

栏    目:Python代码

下一篇:pytorch构建网络模型的4种方法

本文标题:在TensorFlow中屏蔽warning的方式

本文地址:http://www.codeinn.net/misctech/201162.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有