This commit is contained in:
Shimon Turjeman 2023-05-22 17:38:11 -05:00 committed by GitHub
commit d1df335c9a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -481,7 +481,7 @@ def set_tensorflow_log_level(log_level):
Note that tf.Print output are INFO logs, so setting log_level above 0 would hide Note that tf.Print output are INFO logs, so setting log_level above 0 would hide
output from tf.Print. output from tf.Print.
""" """
assert isinstance(log_level, int) and log_level >= 0 and log_level <= 3 assert isinstance(log_level, int) and 0 <= log_level <= 3
os.environ['TF_CPP_MIN_LOG_LEVEL'] = str(log_level) os.environ['TF_CPP_MIN_LOG_LEVEL'] = str(log_level)