Skip to content

Commit 5b254c5

Browse files
author
Ubuntu
committed
update params
1 parent 3428b12 commit 5b254c5

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
@@ -58,7 +58,7 @@ class CustomTrainingArgs(Seq2SeqTrainingArguments):
5858

5959
eval_strategy: str = "epoch"
6060
save_strategy: str = "epoch"
61-
logging_steps: int = 50
61+
logging_steps: int = 1
6262
save_total_limit: int = 2
6363
load_best_model_at_end: bool = True
6464
metric_for_best_model: str = "loss"

0 commit comments

Comments
 (0)