From dabf12eaa6a2798aff332138f177ce948b1b1057 Mon Sep 17 00:00:00 2001 From: meliksahturker Date: Thu, 2 Mar 2023 15:55:45 +0300 Subject: [PATCH] tensorflow warnings are suppressed --- vnlp/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/vnlp/__init__.py b/vnlp/__init__.py index 8ed0e08..ff9045b 100644 --- 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')