首页 PaddleNLP 帖子详情
paddlenlp训练报错
收藏
快速回复
PaddleNLP 问答训练NLP数据 397 0
paddlenlp训练报错
收藏
快速回复
PaddleNLP 问答训练NLP数据 397 0

目前在测试基于UIE开源模型微调训练定制模型提取实体遇到以下问题:
使用环境是 cuda11.6 + paddlenlp 2.6.1 + paddlepaddle-gpu 2.5.1.post116

在使用uie-m-large作为基础模型的时候会报错:KeyError: 'token_type_ids',使用uie-base可以正常训练;

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户