[FEATURE]: ADD LORA+ ALGORITHM

Former-commit-id: a0965cd62c
This commit is contained in:
齐保元
2024-03-13 19:43:27 +08:00
parent 634c44c51a
commit 24c9277488
4 changed files with 130 additions and 3 deletions

View File

@@ -210,6 +210,11 @@ class FinetuningArguments(FreezeArguments, LoraArguments, RLHFArguments, GaloreA
default=False,
metadata={"help": "Whether or not to make only the parameters in the expanded blocks trainable."},
)
# for lora+,[LoRA+: Efficient Low Rank Adaptation of Large Models](https://arxiv.org/pdf/2402.12354.pdf)
lora_lr_ratio: Optional[float] = field(
default=None,
metadata={'help': 'The lora learning_rate ratio of lora_A to lora_B, option:16.0.'},
)
plot_loss: bool = field(
default=False,
metadata={"help": "Whether or not to save the training loss curves."},