paddlenlp训练报错
收藏
目前在测试基于UIE开源模型微调训练定制模型提取实体遇到以下问题:
使用环境是 cuda11.6 + paddlenlp 2.6.1 + paddlepaddle-gpu 2.5.1.post116
在使用uie-m-large作为基础模型的时候会报错:KeyError: 'token_type_ids',使用uie-base可以正常训练;
0
收藏
请登录后评论