在进行情感分类任务时遇到了GPU运行错误，原因是使用预训练模型计算损失时，标签值超过了预期范围。原来的做法是将标签转换为ID，而新方法直接传递整数标签。修正方案是删除不必要的转换步骤，直接返回整数标签，从而解决了问题。
在做情感分类时出现了这个错误。用gpu跑会显示这个，当我切换到cpu跑时，定位到错误label index超范围了。
原因：之前做分类任务的时候是自己用logits算loss的，在做dataset处理的时候convert_tokens_to_ids(label)，然后根据vocab，label就是了一百多。这回我直接将label传入预训练模型里算loss，所以报错了。这里面的label应该是[0,1,2..]
outputs = self.roberta(inputs_embeds=input_emb, attention_mask=attention_mask, labels=label)改正：删掉convert_tokens_to_ids(label)，直接return int(label)，问题解决。
扫一扫
6412
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
