TensorFlow中屏蔽warning的方法

 TensorFlow的日誌級別分爲如下三種:         python

  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出現警告信息,又不想讓警告信息顯示時,可進行以下設置:spa

 

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

1 import os
2 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
相關文章
相關標籤/搜索