728x90
반응형
Disable tensorflow warning message
keras model을 로드하거나 tensorflow를 import 하고 코드를 실행할 때 warning message가 거슬리게 자꾸 터미널에 나타날 때가 있습니다.
코드가 실행이 안되는게 아니라 이런 워닝 메시지가 자꾸 뜹니다. 실행은 잘 되고...
거슬려도 참고 테스트 하다가 이걸 좀 처리를 해줘야겠다! 해서 첫 번째 아래 시도를 해 봤습니다.
import warnings
warnings.filterwarnings(action='ignore')흠.. 주피터 노트북에서 실행할 때는 효과가 있는 듯했지만 역시나 tensrflow 워닝은 잡아주지 못하더군요..
그래서 silence_tensorflow 라는 모듈을 import 했습니다.
> pip install silence_tensorflow
먼저 install을 해주고, 아래와 같이 사용합니다.
from silence_tensorflow import silence_tensorflow
silence_tensorflow()
from keras.models import load_model
그랬더니 아주 깔끔하게 워닝 메시지는 출력하지 않습니다!
모델 하나 가져오려고 하니 워닝이 막막 엄청 많아...
그럼 이만 :-)
728x90
반응형