[trainer] 2 bug fixes and a rename (#12309)

* bug fixes and a rename

* add extended DDP test
This commit is contained in:
Stas Bekman
2021-06-22 11:13:23 -07:00
committed by GitHub
parent 64029abe4c
commit ebe5413589
7 changed files with 112 additions and 18 deletions

View File

@@ -245,7 +245,7 @@ def main():
handlers=[logging.StreamHandler(sys.stdout)],
)
log_level = training_args.get_node_log_level()
log_level = training_args.get_process_log_level()
logger.setLevel(log_level)
datasets.utils.logging.set_verbosity(log_level)
transformers.utils.logging.set_verbosity(log_level)