English | 简体中文
模型 | 结构 | 语言 |
---|---|---|
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 |