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

fix #3116

Browse Source 
Former-commit-id: b7256aa33d761280751518c20f29f9b8ea3fb025
This commit is contained in:
hiyouga
2024-04-03 14:47:59 +08:00
parent 2e2ebc0986
commit 99ed1657db
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},
)