areum

[tensorflow] 로그 레벨 제한 | warning메시지 숨김 본문

Programming/Tensorflow

[tensorflow] 로그 레벨 제한 | warning메시지 숨김

armmy 2023. 1. 27. 16:56
728x90

tensorflow를 사용하면서 로그가 길게 출력되는 경우가 있다.

TF_CPP_MIN_LOG_LEVEL 를 사용하여 로그를 제한할 수 있다.

 

  • 기본값은 0 = 모든 로그가 표시됨.
  • 1 = INFO 로그 표시.
  • 2 = WARNING 로그 표시.
  • 3 = ERROR 로그 표시.

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='숫자'