423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-04 12:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix callback

Browse Source 
Former-commit-id: 1e9401744cadecdef043b6f744b2616a74c64bca
This commit is contained in:
hiyouga 2023-10-15 04:59:44 +08:00
parent 90be167b5f
commit 4f9ca28e11
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/extras/callbacks.py
View File

@ -66,7 +66,7 @@ class LogCallback(TrainerCallback):
self.in_training = True self.in_training = True
self.start_time = time.time() self.start_time = time.time()
self.max_steps = state.max_steps self.max_steps = state.max_steps
if os.path.exists(os.path.join(args.output_dir, LOG_FILE_NAME)): if os.path.exists(os.path.join(args.output_dir, LOG_FILE_NAME)) and args.overwrite_output_dir:
logger.warning("Previous log file in this folder will be deleted.") logger.warning("Previous log file in this folder will be deleted.")
os.remove(os.path.join(args.output_dir, LOG_FILE_NAME)) os.remove(os.path.join(args.output_dir, LOG_FILE_NAME))
@ -135,7 +135,7 @@ class LogCallback(TrainerCallback):
remaining_time=self.remaining_time remaining_time=self.remaining_time
) )
if self.runner is not None: if self.runner is not None:
logger.info("{{'loss': {:.4f}, 'learning_rate': {:.6f}, 'epoch': {:.2f}}}".format( logger.info("{{'loss': {:.4f}, 'learning_rate': {:2.2e}, 'epoch': {:.2f}}}".format(
logs["loss"], logs["learning_rate"], logs["epoch"] logs["loss"], logs["learning_rate"], logs["epoch"]
)) ))