mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-11-02 12:44:20 +08:00
通用信息抽取 UIE模型部署
模型版本说明
支持的模型列表
| 模型 | 结构 | 语言 |
|---|---|---|
uie-base |
12-layers, 768-hidden, 12-heads | 中文 |
uie-medium |
6-layers, 768-hidden, 12-heads | 中文 |
uie-mini |
6-layers, 384-hidden, 12-heads | 中文 |
uie-micro |
4-layers, 384-hidden, 12-heads | 中文 |
uie-nano |
4-layers, 312-hidden, 12-heads | 中文 |
导出部署模型
在部署前，需要先将UIE模型导出成部署模型，导出步骤参考文档导出模型
下载预训练模型
为了方便开发者的测试，下面提供了UIE导出的各模型，开发者可直接下载使用。
其中精度指标F1来源于PaddleNLP中对各模型在互联网自建数据集进行0-shot的实验，详情各参考PaddleNLP中的说明。
| 模型 | 参数大小 | F1 值 |
|---|---|---|
| uie-base | 416M | 78.33 |
| uie-medium | 265M | 78.32 |
| uie-mini | 95M | 72.09 |
| uie-micro | 83M | 66.00 |
| uie-nano | 64M | 62.86 |