TensorFlow中屏蔽warning的方法


 TensorFlow的日志級別分為以下三種:         

  1.  TF_CPP_MIN_LOG_LEVEL = 1         //默認設置,為顯示所有信息                
  2. TF_CPP_MIN_LOG_LEVEL = 2         //只顯示error和warining信息                
  3. TF_CPP_MIN_LOG_LEVEL = 3         //只顯示error信息

所以,當TensorFlow出現警告信息,又不想讓警告信息顯示時,可進行如下設置:

 

python環境下通過在python文件中添加如下兩行代碼,設置TensorFlow日志輸出級別。

1 import os
2 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM