imporve log

This commit is contained in:
hiyouga
2025-01-08 09:56:10 +00:00
parent d23a98825b
commit 47e17dd689
16 changed files with 78 additions and 67 deletions

View File

@@ -239,7 +239,7 @@ class LogCallback(TrainerCallback):
and os.path.exists(os.path.join(args.output_dir, TRAINER_LOG))
and args.overwrite_output_dir
):
logger.warning_once("Previous trainer log in this folder will be deleted.")
logger.warning_rank0_once("Previous trainer log in this folder will be deleted.")
os.remove(os.path.join(args.output_dir, TRAINER_LOG))
@override