fix: remove duplicate lora plus setting
This commit is contained in:
@@ -330,8 +330,6 @@ class TrainerBuilderBase(abc.ABC):
|
||||
training_args_kwargs["run_name"] = None
|
||||
|
||||
# optim/scheduler
|
||||
training_args_kwargs["loraplus_lr_ratio"] = self.cfg.loraplus_lr_ratio
|
||||
training_args_kwargs["loraplus_lr_embedding"] = self.cfg.loraplus_lr_embedding
|
||||
if self.cfg.lr_scheduler in ["one_cycle", "log_sweep", "rex"]:
|
||||
training_args_kwargs["lr_scheduler_type"] = "cosine"
|
||||
training_args_kwargs["alternate_lr_scheduler_type"] = self.cfg.lr_scheduler
|
||||
|
||||
Reference in New Issue
Block a user