[FEATURE]: ADD LORA+ ALGORITHM

Former-commit-id: c35b3c3b1e27171f8a703f88ede1dc8a84c80a56
This commit is contained in:
齐保元
2024-03-13 19:43:27 +08:00
parent 42473ec150
commit 3c91e86268
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."},