加载ernie预训练模型使用存在问题
finley0066 发布于2021-03 浏览:1019 回复:2
0
收藏

MODEL_NAME = "ernie-1.0"

tokenizer = ppnlp.transformers.ErnieTokenizer.from_pretrained(MODEL_NAME)

encoded_text = tokenizer.encode(text="请输入测试样例")

segment_ids = paddle.to_tensor([encoded_text['segment_ids']])

这里的字典encoded_text并无segment_ids键,请问该怎么办?

收藏
点赞
0
个赞
共2条回复 最后由七年期限回复于2021-03
#3七年期限回复于2021-03

今天在paddlenlp群中也有解答 758287592  欢迎加群呀

0
#2七年期限回复于2021-03

安装一下paddlenlp==2.0.0b4  然后重新运行项目

0
TOP
切换版块