fixed TF warnings.

sisinflab · Mar 28, 2024 · 357d554 · 357d554
1 parent e3b031c
commit 357d554
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/ducho/runner/Runner.py b/ducho/runner/Runner.py
@@ -1,4 +1,8 @@
 import os
+# Hiding TensorFlow warnings
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  # or any {'0', '1', '2'}
+# Hiding Transformers' tokenizer warnings
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
 from loguru import logger
 from alive_progress import alive_bar
 import torch
@@ -12,11 +16,6 @@
 from ducho.multimodal.multiple.visual_textual.VisualTextualDataset import VisualTextualDataset
 from ducho.internal.utils.json2dotnotation import banner
 
-# Hiding TensorFlow warnings
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  # or any {'0', '1', '2'}
-# Hiding Transformers' tokenizer warnings
-os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
 
 def camel_case(s):
     return ''.join(x for x in s.title() if not x.isspace()).replace('_', '')