Skip to content

Commit aaee72f

Browse files
committed
docs: show lr and batch size for dpa adapt tuning
1 parent 26fccd1 commit aaee72f

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

doc/dpa_adapt/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@ model = DPAFineTuner(
4040
pretrained="DPA-3.1-3M",
4141
strategy="frozen_head", #"frozen_head" | "finetune"
4242
property_name="homo",
43+
learning_rate=1e-3,
44+
batch_size=512,
4345
)
4446
model.fit(train_data="/data/train", valid_data="/data/valid")
4547

0 commit comments

Comments
 (0)