We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a932b40 commit 3428b12Copy full SHA for 3428b12
src/bart_reddit_lora/train.py
@@ -74,7 +74,7 @@ class CustomTrainingArgs(Seq2SeqTrainingArguments):
74
label_names: List[str] = field(default_factory=lambda: ["labels"])
75
76
# additional custom args
77
- peft_rank: int = field(default=32, metadata={"help": "LoRA adapter rank (r)."})
+ peft_rank: int = field(default=128, metadata={"help": "LoRA adapter rank (r)."})
78
hf_hub_repo_id: str | None = None
79
run_test: bool = field(
80
default=False,
0 commit comments