Skip to content

Commit 10e8d79

Browse files
committed
fix ptuning
1 parent 5196a3e commit 10e8d79

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

autointent/modules/scoring/_ptuning/ptuning.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ class PTuningScorer(BertScorer):
3434
classification_model_config="prajjwal1/bert-tiny",
3535
num_train_epochs=3,
3636
batch_size=8,
37-
task_type="SEQ_CLS",
3837
num_virtual_tokens=10,
3938
seed=42,
4039
training_arguments={"logging_strategy": "no"},

0 commit comments

Comments
 (0)