栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

CUDA error: CUBLAS

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

CUDA error: CUBLAS

在做情感分类时出现了这个错误。用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),问题解决。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/966023.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号