1. 程式人生 > 程式設計 >在TensorFlow中遮蔽warning的方式

在TensorFlow中遮蔽warning的方式

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的方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。