在TensorFlow中如何屏蔽warning

發(fā)布時(shí)間：2021-08-18 15:33:49 來源：億速云閱讀：496 作者：小新欄目：開發(fā)技術(shù)

小編給大家分享一下在TensorFlow中如何屏蔽warning，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

TensorFlow的日志級(jí)別分為以下三種：

TF_CPP_MIN_LOG_LEVEL = 1 //默認(rèn)設(shè)置，為顯示所有信息

TF_CPP_MIN_LOG_LEVEL = 2 //只顯示error和warining信息

TF_CPP_MIN_LOG_LEVEL = 3 //只顯示error信息

所以，當(dāng)TensorFlow出現(xiàn)警告信息，又不想讓警告信息顯示時(shí)，可進(jìn)行如下設(shè)置：

一、python環(huán)境下

通過在python文件中添加如下兩行代碼，設(shè)置TensorFlow日志輸出級(jí)別

impot os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

二、linux c/c++調(diào)用環(huán)境下

在linux環(huán)境下，我們可以通過linux命令，屏蔽掉TensorFlow的warning：

1、暫時(shí)屏蔽warning：

在linux終端中輸入命令：export TF_CPP_MIN_LOG_LEVEL=2，當(dāng)該終端關(guān)閉，既失效；

2、永久屏蔽warning：

修改/etc/profile文件，在該文件中加入export TF_CPP_MIN_LOG_LEVEL=2，但是修改后需要重啟

系統(tǒng)或者使用source /etc/profile命令，使修改后的profile文件生效。

以上是“在TensorFlow中如何屏蔽warning”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！

向AI問一下細(xì)節(jié)

猜你喜歡