Skip to content

GitLab

"tests/utils/test_feature_extraction_utils.py" did not exist on "cb38ffcc5e0ae2fac653342ac36dc75c15ea178f"
Unverified Commit 6bab8368 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

fix logger format for non-main process (#9911)

parent d85691ac
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/seq2seq/finetune_trainer.py
View file @ 6bab8368
...@@ -175,11 +175,11 @@ def main(): ...@@ -175,11 +175,11 @@ def main():
bool(training_args.parallel_mode == ParallelMode.DISTRIBUTED), bool(training_args.parallel_mode == ParallelMode.DISTRIBUTED),
training_args.fp16, training_args.fp16,
) )
transformers.utils.logging.enable_default_handler()
transformers.utils.logging.enable_explicit_format()
# Set the verbosity to info of the Transformers logger (on main process only): # Set the verbosity to info of the Transformers logger (on main process only):
if is_main_process(training_args.local_rank): if is_main_process(training_args.local_rank):
transformers.utils.logging.set_verbosity_info() transformers.utils.logging.set_verbosity_info()
transformers.utils.logging.enable_default_handler()
transformers.utils.logging.enable_explicit_format()
logger.info("Training/evaluation parameters %s", training_args) logger.info("Training/evaluation parameters %s", training_args)
# Set seed # Set seed
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment