Skip to content

Commit 9237515

Browse files
authored
Merge pull request #20 from zhaiwenxi/update-dpa-adapt-readme
Update dpa_adapt README example
2 parents a1d8832 + c6a09ca commit 9237515

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

doc/dpa_adapt/README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,9 @@ model.fit(train_data="/data/train/*", target_key="homo")
3737

3838
# frozen_head / finetune — same interface, different depth
3939
model = DPAFineTuner(
40-
pretrained="DPA-3.1-3M", strategy="frozen_head", property_name="homo"
40+
pretrained="DPA-3.1-3M",
41+
strategy="frozen_head", #"frozen_head" | "finetune"
42+
property_name="homo",
4143
)
4244
model.fit(train_data="/data/train", valid_data="/data/valid")
4345

0 commit comments

Comments
 (0)