Tensorflow, выбор уровня логирования

08.10.2021

По умолчанию Tensorflow (актуально для версии 2.4) выводит все типы сообщений — информационные (INFO), предупреждения (WARNING) и об ошибках (ERROR). Подобное поведение библиотеки довольно часто бывает неудобным.

Одним из способов управления количеством выводимой информации при работе с Tensorflow является присвоение переменной окружения TF_CPP_MIN_LOG_LEVEL нужного уровня логирования до импорта библиотеки:

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 
import tensorflow as tf

Возможные значения для TF_CPP_MIN_LOG_LEVEL и их значение:

  • 0: отображать все сообщения (по умолчанию):
  • 1: не отображать сообщения уровня INFO;
  • 2: не отображать сообщения уровней INFO и WARNING;
  • 3: не отображать INFO, WARNING и ERROR, то есть ничего.