File tree 4 files changed +17
-14
lines changed
examples/multimodal/layoutxlm
paddlenlp/transformers/layoutxlm
4 files changed +17
-14
lines changed Original file line number Diff line number Diff line change 8
8
### 配置环境
9
9
环境依赖
10
10
- cv2
11
- - numpy
11
+ - sentencepiece
12
+ - yacs
12
13
13
14
安装命令:
14
15
``` shell
15
16
pip install opencv-python
16
- pip install numpy
17
+ pip install sentencepiece
18
+ pip install yacs
17
19
```
18
20
19
21
### 数据准备
20
- 处理好的XFUND中文数据集下载地址:https://bj.bcebos.com/v1/paddlenlp/datasets/XFUND.zip
22
+ 处理好的XFUND中文数据集下载地址:https://bj.bcebos.com/v1/paddlenlp/datasets/XFUND.zip 。
23
+
24
+ 下载并解压该数据集,解压后将数据集放置在当前目录下。
21
25
22
26
### 执行Fine-tuning
23
27
1 . `` Semantic Entity Recognition `` 任务启动Fine-tuning的方式如下:
@@ -28,7 +32,6 @@ pip install numpy
28
32
# best metrics: {'precision': 0.8514686248331108, 'recall': 0.9354602126879354, 'f1': 0.8914904770225406}
29
33
```
30
34
31
-
32
35
2. ` ` Relation Extraction` ` 任务启动Fine-tuning的方式如下:
33
36
` ` ` shell
34
37
bash run_xfun_re.sh
Original file line number Diff line number Diff line change @@ -3,10 +3,10 @@ export CUDA_VISIBLE_DEVICES=0
3
3
python ./run_xfun_re.py \
4
4
--model_name_or_path " layoutxlm-base-uncased" \
5
5
--max_seq_length 512 \
6
- --train_data_dir " zh.train/img " \
7
- --train_label_path " zh.train /xfun_normalize_train.json" \
8
- --eval_data_dir " zh.val/img " \
9
- --eval_label_path " zh.val /xfun_normalize_val.json" \
6
+ --train_data_dir " XFUND/zh_train/image " \
7
+ --train_label_path " XFUND/zh_train /xfun_normalize_train.json" \
8
+ --eval_data_dir " XFUND/zh_val/image " \
9
+ --eval_label_path " XFUND/zh_val /xfun_normalize_val.json" \
10
10
--num_train_epochs 200 \
11
11
--eval_steps 50 \
12
12
--save_steps 500 \
Original file line number Diff line number Diff line change @@ -3,10 +3,10 @@ export CUDA_VISIBLE_DEVICES=0
3
3
python ./run_xfun_ser.py \
4
4
--model_name_or_path " layoutxlm-base-uncased" \
5
5
--max_seq_length 512 \
6
- --train_data_dir " zh.train/img " \
7
- --train_label_path " zh.train /xfun_normalize_train.json" \
8
- --eval_data_dir " zh.val/img " \
9
- --eval_label_path " zh.val /xfun_normalize_val.json" \
6
+ --train_data_dir " XFUND/zh_train/image " \
7
+ --train_label_path " XFUND/zh_train /xfun_normalize_train.json" \
8
+ --eval_data_dir " XFUND/zh_val/image " \
9
+ --eval_label_path " XFUND/zh_val /xfun_normalize_val.json" \
10
10
--num_train_epochs 200 \
11
11
--eval_steps 10 \
12
12
--save_steps 500 \
Original file line number Diff line number Diff line change @@ -32,8 +32,8 @@ def read_config(fp=None):
32
32
dir_name = os .path .dirname (os .path .abspath (__file__ ))
33
33
fp = os .path .join (dir_name , "visual_backbone.yaml" )
34
34
with open (fp , "r" ) as fin :
35
- yacs = try_import ("yacs" )
36
- cfg = yacs . config .CfgNode ().load_cfg (fin )
35
+ yacs_config = try_import ("yacs.config " )
36
+ cfg = yacs_config .CfgNode ().load_cfg (fin )
37
37
cfg .freeze ()
38
38
return cfg
39
39
You can’t perform that action at this time.
0 commit comments