ernie能使用CPM-LM的数据吗?
windflee 发布于2021-06 浏览:3641 回复:1
1
收藏

ernie能使用CPM-LM的数据吗? 看到有在hub中用GPT2使用CPM-LM的数据,但是没有ernie使用CPM-LM的示例。

收藏
点赞
1
个赞
共1条回复 最后由用户已被禁言回复于2022-03
#2回复于2021-09

#paddlenlp>=2.0.0rc
from paddlenlp.transformers.gpt.modeling import GPTForPretraining
from paddlenlp.transformers.gpt.tokenizer import GPTChineseTokenizer

# 加载 PaddleNLP 自带的预训练中文 GPT2 模型
model = GPTForPretraining.from_pretrained('gpt-cpm-large-cn')

# 设置模型为评估状态
model.eval()

# 加载编解码器
tokenizer = GPTChineseTokenizer.from_pretrained('gpt-cpm-large-cn')

1
快速回复
TOP
切换版块