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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章