From ddee718b31a5bf3cb39c5adf3f8e0be8fddf9dbb Mon Sep 17 00:00:00 2001
From: liu-zichen <liuzichen@dbis.nankai.edu.cn>
Date: Tue, 13 Aug 2024 16:33:34 +0800
Subject: [PATCH] fix lr not change

---
 src/llamafactory/train/ppo/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/llamafactory/train/ppo/trainer.py b/src/llamafactory/train/ppo/trainer.py
index 58ea83d8..3b792061 100644
--- 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