fix

modelscope · Dec 24, 2024 · a1e9876 · a1e9876
1 parent 92233d3
commit a1e9876
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/train/plugins/train_cls.sh b/examples/train/plugins/train_cls.sh
@@ -8,6 +8,7 @@ swift sft \
     --dataset simpleai/HC3-Chinese:baike_cls#1000 \
     --num_train_epochs 1 \
     --num_labels 2 \
+    --task_type seq_cls \
     --per_device_train_batch_size 1 \
     --learning_rate 1e-5 \
     --gradient_accumulation_steps 16 \