From a5dbd8bb74a494caafa2801b03e38e57d6a08ac4 Mon Sep 17 00:00:00 2001 From: WenmuZhou <572459439@qq.com> Date: Sat, 18 Dec 2021 04:29:53 +0000 Subject: [PATCH] rm unsused code --- ppstructure/vqa/README.md | 12 +----------- ppstructure/vqa/train_ser.py | 2 +- 2 files changed, 2 insertions(+), 12 deletions(-) diff --git a/ppstructure/vqa/README.md b/ppstructure/vqa/README.md index 08db718f..744cdc74 100644 --- a/ppstructure/vqa/README.md +++ b/ppstructure/vqa/README.md @@ -182,16 +182,6 @@ python3 eval_ser.py \ ``` 最终会打印出`precision`, `recall`, `f1`等指标 - -```shell -export CUDA_VISIBLE_DEVICES=0 -python3.7 infer_ser.py \ - --model_name_or_path "./PP-Layout_v1.0_ser_pretrained/" \ - --output_dir "output_res/" \ - --infer_imgs "XFUND/zh_val/image/" \ - --ocr_json_path "XFUND/zh_val/xfun_normalize_val.json" -``` - * 使用评估集合中提供的OCR识别结果进行预测 ```shell @@ -237,7 +227,7 @@ python3 train_re.py \ --eval_data_dir "XFUND/zh_val/image" \ --eval_label_path "XFUND/zh_val/xfun_normalize_val.json" \ --label_map_path 'labels/labels_ser.txt' \ - --num_train_epochs 2 \ + --num_train_epochs 200 \ --eval_steps 10 \ --output_dir "output/re/" \ --learning_rate 5e-5 \ diff --git a/ppstructure/vqa/train_ser.py b/ppstructure/vqa/train_ser.py index a722f921..df072480 100644 --- a/ppstructure/vqa/train_ser.py +++ b/ppstructure/vqa/train_ser.py @@ -139,7 +139,7 @@ def train(args): global_step = 0 tr_loss = 0.0 - set_seed(ags.seed) + set_seed(args.seed) best_metrics = None train_reader_cost = 0.0 -- GitLab