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 osos.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的方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答