tensorflow warnings are suppressed

vngrs-ai · Mar 2, 2023 · dabf12e · dabf12e
1 parent aaeae7a
commit dabf12e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/vnlp/__init__.py b/vnlp/__init__.py
@@ -1,3 +1,7 @@
+import os
+# Suppress tensorflow warnings
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
+
 import tensorflow as tf
 # Prevent tensorflow from allocating whole GPU memory.
 gpus = tf.config.experimental.list_physical_devices('GPU')