Отключить дополнительную информацию о предупреждении Tensorflow - PullRequest
0 голосов
/ 16 июня 2020

У меня проблемы с предупреждениями tenorflow os. Я знаю, что существует довольно много решений по переполнению стека, и уже пробовал все это. Как, например, это: Отключить отладочную информацию Tensorflow

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
import logging
logging.getLogger('tensorflow').setLevel(logging.ERROR)

import tensorflow as tf
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
tf.config.set_soft_device_placement(True)
tf.debugging.set_log_device_placement(True)
list_gpu = tf.config.experimental.list_physical_devices(device_type='GPU')
for gpu in list_gpu: tf.config.experimental.set_memory_growth(gpu, True)
print('Num GPUs Available: ', len(list_gpu))

Как напечатать это:

Executing op RandomUniform in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op Sub in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op Mul in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op Add in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op VarHandleOp in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op VarIsInitializedOp in device /job:localhost/replica:0/task:0/device:GPU:0
Executing op LogicalNot in device /job:localhost/replica:0/task:0/device:GPU:0

Я что-то здесь упустил?

  • Версия Tensorflow = 2.2.1
  • Использование Google Colab

Пожалуйста, помогите! TIA.

1 Ответ

0 голосов
/ 16 июня 2020

Вы пробовали

tf.logging.set_verbosity(tf.logging.ERROR)

или

tf.get_logger().setLevel(3)
...