CUDA error: device-side assert triggered
【摘要】
CUDA error: device-side assert triggered
这个报错,错误不一定在报错的地方,可能在报错的前面代码出错了。
原因:
1.实际类别数和网络输出的类别数不一致,比如标签出现-1,超过分类个数的值。
2.我这是因为没有归一化,loss为nan了。
这个错误的关键在于错误并...
CUDA error: device-side assert triggered
这个报错,错误不一定在报错的地方,可能在报错的前面代码出错了。
原因:
1.实际类别数和网络输出的类别数不一致,比如标签出现-1,超过分类个数的值。
2.我这是因为没有归一化,loss为nan了。
这个错误的关键在于错误并不在出错的哪一行。
看了一些网上的帖子和自己错误,错误的原因是用index超过了tensor的范围。
-
a = torch.randn(10)
-
a[11]
文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。
原文链接:blog.csdn.net/jacke121/article/details/104792239
【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)