2023-02-10 05:32:21 +00:00
|
|
|
PYTHONPATH=. python examples/pytorch/text_classification/finetune_text_classification.py \
|
2023-05-16 14:32:43 +08:00
|
|
|
--task 'text-classification' \
|
2023-02-10 05:32:21 +00:00
|
|
|
--model 'damo/nlp_structbert_backbone_base_std' \
|
2023-05-13 12:12:04 +08:00
|
|
|
--train_dataset_name 'clue' \
|
2023-06-08 19:36:05 +08:00
|
|
|
--val_dataset_name 'clue' \
|
2023-05-13 12:12:04 +08:00
|
|
|
--train_subset_name 'tnews' \
|
2023-06-08 19:36:05 +08:00
|
|
|
--val_subset_name 'tnews' \
|
|
|
|
|
--train_split 'train' \
|
|
|
|
|
--val_split 'validation' \
|
2023-02-10 05:32:21 +00:00
|
|
|
--first_sequence 'sentence' \
|
2023-06-08 19:36:05 +08:00
|
|
|
--label label \
|
2023-02-10 05:32:21 +00:00
|
|
|
--preprocessor 'sen-cls-tokenizer' \
|
2023-05-16 14:32:43 +08:00
|
|
|
--use_model_config True \
|
|
|
|
|
--max_epochs 1 \
|
2023-06-08 19:36:05 +08:00
|
|
|
--per_device_train_batch_size 16 \
|
|
|
|
|
--per_device_eval_batch_size 16 \
|
|
|
|
|
--eval_interval 100 \
|
|
|
|
|
--eval_strategy by_step \
|
|
|
|
|
--work_dir './tmp' \
|
|
|
|
|
--train_data_worker 0 \
|
|
|
|
|
--eval_data_worker 0 \
|
|
|
|
|
--lr 1e-5 \
|
2023-05-16 14:32:43 +08:00
|
|
|
--eval_metrics 'seq-cls-metric' \
|