模型:

nghuyong/ernie-3.0-base-zh

英文

ERNIE-3.0-base-zh

介绍

ERNIE 3.0:用于语言理解和生成的大规模知识增强预训练模型详情请见 https://arxiv.org/abs/2107.02137

发布模型信息

此发布的pytorch模型是由官方发布的PaddlePaddle ERNIE模型转换而来,并进行了一系列实验来检查转换的准确性。

使用方法

然后您可以像以前一样加载ERNIE-3.0模型:

from transformers import BertTokenizer, ErnieForMaskedLM

tokenizer = BertTokenizer.from_pretrained("nghuyong/ernie-3.0-base-zh")
model = ErnieForMaskedLM.from_pretrained("nghuyong/ernie-3.0-base-zh")

引用

@article{sun2021ernie,
  title={Ernie 3.0: Large-scale knowledge enhanced pre-training for language understanding and generation},
  author={Sun, Yu and Wang, Shuohuan and Feng, Shikun and Ding, Siyu and Pang, Chao and Shang, Junyuan and Liu, Jiaxiang and Chen, Xuyi and Zhao, Yanbin and Lu, Yuxiang and others},
  journal={arXiv preprint arXiv:2107.02137},
  year={2021}
}