加载ernie预训练模型使用存在问题
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
个赞
请登录后评论
TOP
切换版块
今天在paddlenlp群中也有解答 758287592 欢迎加群呀
安装一下paddlenlp==2.0.0b4 然后重新运行项目