Tensorflow:log设置方式(python和c++)

从源码编译tensorflow应该都知道,tensorflow的源码是c++写的,通过bazel编译,python接口的底层源码也是c++。

所以LZ也研究了下tensorflow的LOG设置,在logging.h中头文件的定义如下所示:

const int INFO = 0;            // base_logging::INFO;
const int WARNING = 1;         // base_logging::WARNING;
const int ERROR = 2;           // base_logging::ERROR;
const int FATAL = 3;           // base_logging::FATAL;
const int NUM_SEVERITIES = 4;  // base_logging::NUM_SEVERITIES;

主要是有四个LOG级别,分别为INFO, WARNING, ERROR和FATAL,是四个!

在python接口使用是非常便捷的,直接使用下面代码进行设置

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

那么在c++中怎么进行设置呢?

根据之前的环境变量那篇博客,可以选择一次性的设置,当终端关闭后失效或者长期有效。

下面就介绍一次性设置的方式,打开终端,输入下方代码,在不进行设置的时候,默认为0

export TF_CPP_MIN_LOG_LEVEL=1

运行你的TF的时候就干干净净(≧▽≦)/啦啦啦!Merry Christmas!

发布了305 篇原创文章 · 获赞 208 · 访问量 60万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章