Merge pull request #5163 from liu-zichen/fix_ppo_optim

fix lr not change
This commit is contained in:
hoshi-hiyouga authored and GitHub committed 2024-08-19 23:56:24 +08:00
commit da335d42c3
1 file changed
+1
+1
View File
@@ -135,6 +135,7 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
dataset=train_dataset,
data_collator=data_collator,
lr_scheduler=scheduler,
optimizer=optimizer,
)
self.args = training_args