TensorFlow中如何有效屏蔽或忽略特定warning信息？

本文共计272个文字，预计阅读时间需要2分钟。

TensorFlow的日志级别分为以下三种：- `TF_CPP_MIN_LOG_LEVEL=1` // 默认设置，显示所有信息- `TF_CPP_MIN_LOG_LEVEL=2` // 只显示error和warning信息- `TF_CPP_MIN_LOG_LEVEL=3` // 只显示error信息

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的方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持易盾网络。