Skip to content

Commit 3428b12

Browse files
author
Ubuntu
committed
updated hyper p arms
1 parent a932b40 commit 3428b12

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/bart_reddit_lora/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ class CustomTrainingArgs(Seq2SeqTrainingArguments):
7474
label_names: List[str] = field(default_factory=lambda: ["labels"])
7575

7676
# additional custom args
77-
peft_rank: int = field(default=32, metadata={"help": "LoRA adapter rank (r)."})
77+
peft_rank: int = field(default=128, metadata={"help": "LoRA adapter rank (r)."})
7878
hf_hub_repo_id: str | None = None
7979
run_test: bool = field(
8080
default=False,

0 commit comments

Comments
 (0)