We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 26fccd1 + aaee72f commit 8e7c1ffCopy full SHA for 8e7c1ff
1 file changed
doc/dpa_adapt/README.md
@@ -40,6 +40,8 @@ model = DPAFineTuner(
40
pretrained="DPA-3.1-3M",
41
strategy="frozen_head", #"frozen_head" | "finetune"
42
property_name="homo",
43
+ learning_rate=1e-3,
44
+ batch_size=512,
45
)
46
model.fit(train_data="/data/train", valid_data="/data/valid")
47
0 commit comments