mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-04 04:32:50 +08:00
Merge pull request #2525 from stephen-nju/main
update project_kwargs for ppo config Former-commit-id: 4aab19c7ef160b18eac5c6aea5e648cabd2db72c
This commit is contained in:
commit
db53b67fe4
@ -61,6 +61,7 @@ def run_ppo(
|
||||
use_score_norm=finetuning_args.ppo_score_norm,
|
||||
whiten_rewards=finetuning_args.ppo_whiten_rewards,
|
||||
accelerator_kwargs={"step_scheduler_with_optimizer": False},
|
||||
project_kwargs={"logging_dir": training_args.logging_dir},
|
||||
)
|
||||
|
||||
# Create optimizer and scheduler
|
||||
|
Loading…
x
Reference in New Issue
Block a user