fix lr not change

Former-commit-id: ddee718b31a5bf3cb39c5adf3f8e0be8fddf9dbb
2025-08-23 06:12:50 +08:00 · 2024-08-13 16:33:34 +08:00 · 2024-08-13 16:33:34 +08:00 · 8a7ab8ab21
commit 8a7ab8ab21
parent 753cb0f9b6
1 changed files with 1 additions and 0 deletions
--- a/src/llamafactory/train/ppo/trainer.py
+++ b/src/llamafactory/train/ppo/trainer.py
@ -135,6 +135,7 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
            dataset=train_dataset,
            data_collator=data_collator,
            lr_scheduler=scheduler,
+            optimizer=optimizer,
        )

        self.args = training_args