在TensorFlow中屏蔽warning的方式
TensorFlow的日志級別分為以下三種:
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日志輸出級別
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)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
淺談pandas篩選出表中滿足另一個(gè)表所有條件的數(shù)據(jù)方法
今天小編就為大家分享一篇淺談pandas篩選出表中滿足另一個(gè)表所有條件的數(shù)據(jù)方法,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-02-02PyQt教程之自定義組件Switch?Button的實(shí)現(xiàn)
這篇文章主要為大家詳細(xì)介紹了PyQt中如何實(shí)現(xiàn)自定義組件Switch?Button,文中的示例代碼簡潔易懂,具有一定的學(xué)習(xí)價(jià)值,感興趣的可以了解一下2023-05-05利用keras加載訓(xùn)練好的.H5文件,并實(shí)現(xiàn)預(yù)測圖片
今天小編就為大家分享一篇利用keras加載訓(xùn)練好的.H5文件,并實(shí)現(xiàn)預(yù)測圖片,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01編寫Python腳本抓取網(wǎng)絡(luò)小說來制作自己的閱讀器
這篇文章主要介紹了編寫Python腳本抓取網(wǎng)絡(luò)小說來制作自己的閱讀器的方法,包括對小說的章節(jié)排列等方面的優(yōu)化,對于Python學(xué)習(xí)者來說非常具有實(shí)踐意義!需要的朋友可以參考下2015-08-08Python中輸出ASCII大文字、藝術(shù)字、字符字小技巧
這篇文章主要介紹了Python中輸出ASCII大文字、藝術(shù)字、字符字小技巧,本文首先給出了ASCII大文字、藝術(shù)字、字符字的圖片效果,然后給出了Python中的實(shí)現(xiàn)方法,需要的朋友可以參考下2015-04-04