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

fix #3116

Browse Source
This commit is contained in:
hiyouga
2024-04-03 14:47:59 +08:00
parent f0a9245c7e
commit ce77d98872
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/train/ppo/trainer.py
View File

@@ -66,7 +66,7 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
use_score_norm=finetuning_args.ppo_score_norm,
whiten_rewards=finetuning_args.ppo_whiten_rewards,
accelerator_kwargs={"step_scheduler_with_optimizer": False},
log_with=training_args.report_to[0] if training_args.report_to is not None else None,
log_with=training_args.report_to[0] if training_args.report_to else None,
project_kwargs={"logging_dir": training_args.logging_dir},
)