Update TF_CPP_MIN_LOG_LEVEL

deliahu · deliahu · commit 1f60032e96b6 · 2019-08-08T21:53:22.000-07:00
(cherry picked from commit dcf8ed9)
diff --git a/pkg/workloads/cortex/lib/tf_lib.py b/pkg/workloads/cortex/lib/tf_lib.py
@@ -31,9 +31,12 @@
 }
 
 
+CPP_LOG_LEVEL = {"DEBUG": "1", "INFO": "2", "WARN": "3", "ERROR": "4", "FATAL": "5"}
+
+
 def set_logging_verbosity(verbosity):
     tf.logging.set_verbosity(verbosity)
-    os.environ["TF_CPP_MIN_LOG_LEVEL"] = str(tf.logging.__dict__[verbosity] / 10)
+    os.environ["TF_CPP_MIN_LOG_LEVEL"] = CPP_LOG_LEVEL[verbosity]
 
 
 def get_column_tf_types(model_name, ctx, training=True):
