Tesnsorflow設置日誌顯示級別

Tensorflow運行時經常出現大量的warning信息及設備信息等,如

2019-01-08 15:01:50.069717: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 AVX512F FMA
2019-01-08 15:01:52.378312: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1405] Found device 0 with properties: 
name: Tesla P100-PCIE-16GB major: 6 minor: 0 memoryClockRate(GHz): 1.3285

設置日誌顯示級別可以去掉這些信息。

Tensorflow日誌級別分爲三種:

import os
os.environ["TF_CPP_MIN_LOG_LEVEL"]='1' # 這是默認的顯示等級,顯示所有信息
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只顯示 warning 和 Error 
os.environ["TF_CPP_MIN_LOG_LEVEL"]='3' # 只顯示 Error  

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章