Fix UIE README (PaddlePaddle#3665)

* fix uie reamde * test=document_fix
lijing89 · Nov 3, 2022 · 02b4383 · 02b4383
1 parent 6471ce3
commit 02b4383
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/model_zoo/uie/README.md b/model_zoo/uie/README.md
@@ -650,6 +650,7 @@ python finetune.py  \
     --per_device_train_batch_size  16 \
     --num_train_epochs 100 \
     --learning_rate 1e-5 \
+    --label_names 'start_positions' 'end_positions' \
     --do_train \
     --do_eval \
     --do_export \
@@ -686,6 +687,7 @@ python -u -m paddle.distributed.launch --gpus "0,1" finetune.py \
     --do_eval \
     --do_export \
     --export_model_dir $finetuned_model \
+    --label_names 'start_positions' 'end_positions' \
     --overwrite_output_dir \
     --disable_tqdm True \
     --metric_for_best_model eval_f1 \