diff --git a/ducho/runner/Runner.py b/ducho/runner/Runner.py index e9207ae..3dbbd54 100644 --- 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('_', '')