SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Stas Bekman
2021-02-01 00:08:12 -08:00
committed by GitHub
parent d85691ac75
commit 6bab83683b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/seq2seq/finetune_trainer.py
View File

@@ -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