24 lines
654 B
Bash
24 lines
654 B
Bash
python finetune.py \
|
|
--device gpu \
|
|
--logging_steps 10 \
|
|
--eval_steps 2000 \
|
|
--save_steps 2000 \
|
|
--seed 1000 \
|
|
--model_name_or_path /home/workplace/models/uie-base \
|
|
--output_dir ./checkpoint \
|
|
--train_path ./data/train.txt \
|
|
--dev_path ./data/dev.txt \
|
|
--max_seq_len 256 \
|
|
--per_device_train_batch_size 4 \
|
|
--per_device_eval_batch_size 8 \
|
|
--num_train_epochs 10 \
|
|
--learning_rate 1e-5 \
|
|
--do_train \
|
|
--do_eval \
|
|
--do_export \
|
|
--overwrite_output_dir \
|
|
--disable_tqdm True \
|
|
--metric_for_best_model eval_f1 \
|
|
--load_best_model_at_end True \
|
|
--save_total_limit 1
|