423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-16 00:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix lr not change

Browse Source 
Former-commit-id: 387dd2d51b5d8cd666459040fdd16525b34720d9
This commit is contained in:
liu-zichen 2024-08-13 16:33:34 +08:00
parent 2e257d6af0
commit 75dfe259cf
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/llamafactory/train/ppo/trainer.py
View File

@ -135,6 +135,7 @@ class CustomPPOTrainer(PPOTrainer, Trainer):
dataset=train_dataset, dataset=train_dataset,
data_collator=data_collator, data_collator=data_collator,
lr_scheduler=scheduler, lr_scheduler=scheduler,
optimizer=optimizer,
) )
self.args = training_args self.args = training_args