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日志輸出級別。
1 import os 2 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"