Skip to content

Commit 3d2b9c2

Browse files
committed
添加tplinker_plus在duie_v1的结果
1 parent 646f478 commit 3d2b9c2

File tree

5 files changed

+427
-9
lines changed

5 files changed

+427
-9
lines changed

.gitignore

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,6 @@ coverage.xml
5656
*.pot
5757

5858
# Django stuff:
59-
*.log
6059
local_settings.py
6160
db.sqlite3
6261
db.sqlite3-journal
@@ -126,4 +125,5 @@ venv.bak/
126125
dmypy.json
127126

128127
# Pyre type checker
129-
.pyre/
128+
.pyre/
129+
outputs

README.md

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,17 @@ GPLinker_pytorch
88
- 其中`TPLinker_Plus`代码在模型部分可能有点区别。
99

1010
# 更新
11+
- 2022/03/01 添加`tplinker_plus``duie_v1`上的结果。
1112
- 2022/02/25 现已在Dev分支更新最新的huggingface全家桶版本的代码,main分支是之前旧的代码(执行效率慢)
1213

14+
# 结果
15+
Tips: `gplinker``RTX3090`条件下要训练`5-6h`
16+
| method | pretrained_model_name_or_path | f1 | precision | recall |
17+
| ------------- | ----------------------------- | ------------------ | ------------------ | ------------------ |
18+
| gplinker | hfl/chinese-roberta-wwm-ext | 0.8214065255731926 | 0.8250077498782166 | 0.8178366038895478 |
19+
| gplinker | bert-base-chinese | 0.8198087178424598 | 0.8146470447994109 | 0.8250362175688137 |
20+
| tplinker_plus | bert-base-chinese | 0.8202398259785976 | 0.8169624387588463 | 0.8235436147328684 |
21+
1322
# 依赖
1423
所需的依赖如下:
1524
- fastcore==1.3.29
@@ -68,13 +77,6 @@ accelerate launch train.py \
6877
- `num_workers`: `dataloader``num_workers`参数,`linux`系统下发现`GPU`使用率不高的时候可以尝试设置这个参数大于`0`,而`windows`下最好设置为`0`,不然会报错。
6978

7079

71-
# 结果
72-
Tips: `gplinker``RTX3090`条件下要训练`5-6h`
73-
| method | pretrained_model_name_or_path | f1 | precision | recall |
74-
| -------- | ----------------------------- | ------------------ | ------------------ | ------------------ |
75-
| gplinker | hfl/chinese-roberta-wwm-ext | 0.8214065255731926 | 0.8250077498782166 | 0.8178366038895478 |
76-
| gplinker | bert-base-chinese | 0.8198087178424598 | 0.8146470447994109 | 0.8250362175688137 |
77-
7880
# Tensorboard日志
7981
<p align="center">
8082
<img src="figure/tensorboard_log.jpg" width="100%" />
File renamed without changes.

0 commit comments

Comments
 (0)