fix lr not change

This commit is contained in:
liu-zichen committed 2024-08-13 16:33:34 +08:00
1 parent 625a0e32c4
commit ddee718b31
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