Update TF_CPP_MIN_LOG_LEVEL

(cherry picked from commit dcf8ed9)
cortexlabs · Aug 9, 2019 · 1f60032 · 1f60032
1 parent f766610
commit 1f60032
Showing 1 changed file with 4 additions and 1 deletion.
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):