在进行信息提取实验的过程中,数据标注应该没问题,选择了PaddleNLP的uie-base模型进行训练,并进行模型微调,官方文档用的是gpu,但是由于电脑没有gpu,因此我改成了cpu,设置的epoch是20次,但是训练过了20次依然不停,请问一下有没有知道的伙伴可以解惑一下,下边时模型微调时的代码:
python finetune.py \
--device cpu \
--logging_steps 10 \
--save_steps 100 \
--eval_steps 100 \
--seed 1000 \
--model_name_or_path uie-base \
--output_dir ./checkpoint/model_best \
--train_path data/train.txt \
--dev_path data/dev.txt \
--max_seq_len 512 \
--per_device_train_batch_size 16 \
--per_device_eval_batch_size 16 \
--num_train_epochs 20 \
--learning_rate 1e-5 \
--do_train \
--do_eval \
--do_export \
--export_model_dir ./checkpoint/model_best \
--overwrite_output_dir \
--disable_tqdm True \
--metric_for_best_model eval_f1 \
--load_best_model_at_end True \
--save_total_limit 1
下面是命令行的mo'ximoxi模型微调结果: