Updated TF logging error

rish-16 · Aug 8, 2019 · bd5c485 · bd5c485
1 parent ce6ac5b
commit bd5c485
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/gpt2_client/__pycache__/__init__.cpython-37.pyc b/gpt2_client/__pycache__/__init__.cpython-37.pyc
diff --git a/gpt2_client/__pycache__/gpt2_client.cpython-37.pyc b/gpt2_client/__pycache__/gpt2_client.cpython-37.pyc
diff --git a/gpt2_client/gpt2_client.py b/gpt2_client/gpt2_client.py
@@ -409,7 +409,7 @@ def cond(*args):
         return tokens
 
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
-tf.logging.set_verbosity(tf.logging.ERROR)
+tf.logging.set_verbosity(tf.compat.v1.logging.ERROR)
 
 def default_hparams():
     return HParams(