Setup logging with a stdout handler (#9816)

This commit is contained in:
Sylvain Gugger
2021-01-27 03:39:11 -05:00
committed by GitHub
parent 2c891c156d
commit f2fabedbab
11 changed files with 20 additions and 9 deletions

View File

@@ -188,8 +188,9 @@ def main():
logging.basicConfig(
format="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
datefmt="%m/%d/%Y %H:%M:%S",
level=logging.INFO if is_main_process(training_args.local_rank) else logging.WARN,
handlers=[logging.StreamHandler(sys.stdout)],
)
logger.setLevel(logging.INFO if is_main_process(training_args.local_rank) else logging.WARN)
# Log on each process the small summary:
logger.warning(