File tree Expand file tree Collapse file tree 5 files changed +427
-9
lines changed Expand file tree Collapse file tree 5 files changed +427
-9
lines changed Original file line number Diff line number Diff line change @@ -56,7 +56,6 @@ coverage.xml
56
56
* .pot
57
57
58
58
# Django stuff:
59
- * .log
60
59
local_settings.py
61
60
db.sqlite3
62
61
db.sqlite3-journal
@@ -126,4 +125,5 @@ venv.bak/
126
125
dmypy.json
127
126
128
127
# Pyre type checker
129
- .pyre /
128
+ .pyre /
129
+ outputs
Original file line number Diff line number Diff line change @@ -8,8 +8,17 @@ GPLinker_pytorch
8
8
- 其中` TPLinker_Plus ` 代码在模型部分可能有点区别。
9
9
10
10
# 更新
11
+ - 2022/03/01 添加` tplinker_plus ` 在` duie_v1 ` 上的结果。
11
12
- 2022/02/25 现已在Dev分支更新最新的huggingface全家桶版本的代码,main分支是之前旧的代码(执行效率慢)
12
13
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
+
13
22
# 依赖
14
23
所需的依赖如下:
15
24
- fastcore==1.3.29
@@ -68,13 +77,6 @@ accelerate launch train.py \
68
77
- ` num_workers ` : ` dataloader ` 的` num_workers ` 参数,` linux ` 系统下发现` GPU ` 使用率不高的时候可以尝试设置这个参数大于` 0 ` ,而` windows ` 下最好设置为` 0 ` ,不然会报错。
69
78
70
79
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
-
78
80
# Tensorboard日志
79
81
<p align =" center " >
80
82
<img src="figure/tensorboard_log.jpg" width="100%" />
File renamed without changes.
File renamed without changes.
You can’t perform that action at this time.
0 commit comments