CODEWITHSUNDEEP
pythontensorflowkeras
Cole
Cole

Reputation: 399

How to disable all tensorflow warnings?

I have a for loop with several different deep learning models in it that generates this warning:

WARNING:tensorflow:5 out of the last 5 calls to <function Model.make_predict_function.<locals>.predict_function at 0x000001B0A8CC90D0> triggered tf.function retracing. Tracing is expensive and the excessive number of tracings could be due to (1) creating @tf.function repeatedly in a loop, (2) passing tensors with different shapes, (3) passing Python objects instead of tensors. For (1), please define your @tf.function outside of the loop. For (2), @tf.function has experimental_relax_shapes=True option that relaxes argument shapes that can avoid unnecessary retracing. For (3), please refer to https://www.tensorflow.org/guide/function#controlling_retracing and https://www.tensorflow.org/api_docs/python/tf/function for  more details.
WARNING:tensorflow:6 out of the last 6 calls to <function Model.make_predict_function.<locals>.predict_function at 0x000001B0A6C01940> triggered tf.function retracing. Tracing is expensive and the excessive number of tracings could be due to (1) creating @tf.function repeatedly in a loop, (2) passing tensors with different shapes, (3) passing Python objects instead of tensors. For (1), please define your @tf.function outside of the loop. For (2), @tf.function has experimental_relax_shapes=True option that relaxes argument shapes that can avoid unnecessary retracing. For (3), please refer to https://www.tensorflow.org/guide/function#controlling_retracing and https://www.tensorflow.org/api_docs/python/tf/function for  more details.

I have tried many different things in the for loop to stop it from popping up with no success. Is there simply a way to disable all warnings?

Upvotes: 6

Views: 12641

Answers (3)

s510
s510

Reputation: 2832

Just set tensorflow logger to ERROR:

import tensorflow as tf
tf.get_logger().setLevel('ERROR')

Upvotes: 4

Kapil Musale
Kapil Musale

Reputation: 223

You can use this to avoid warnings:

import os
import tensorflow as tf

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

Upvotes: 8

Sibtain Reza
Sibtain Reza

Reputation: 533

Use this:

tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)

Upvotes: 8

Related Questions