ERNIE-GEN模型汇总#

下表汇总介绍了目前PaddleNLP支持的ERNIE-GEN模型对应预训练权重。 关于模型的具体细节可以参考对应链接。

Pretrained Weight

Language

Details of the model

ernie-gen-base-en

English

12-layer, 768-hidden, 12-heads, 108M parameters. Trained on lower-cased English text.

ernie-gen-large-en

English

24-layer, 1024-hidden, 16-heads, 336M parameters. Trained on lower-cased English text.

ernie-gen-large-en-430g

English

24-layer, 1024-hidden, 16-heads, 336M parameters. Trained on lower-cased English text. with extended data (430 GB).