Skip to content

Latest commit

 

History

History
42 lines (28 loc) · 1.77 KB

README_CN.md

File metadata and controls

42 lines (28 loc) · 1.77 KB

English | 简体中文

通用信息抽取 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

详细部署文档