RuntimeError: CUDA error: device-side assert triggered

作者：花生_TL007 | 2024-02-16 01:06:13

踩

runtimeerror: cuda error: device-side assert triggered

every blog every motto: Just live your life cause we don’t live twice.

0. 前言

c_loss = nn.CrossEntropyLoss()
loss0 = c_loss(d0, labels_v.long())
1
2

报错的原因是，在pytorch中做损失函数计算时，标签为（batch,height,width），如果类别为10类，那么其中的值应该为 0~9，即：
0<= value<=C-1,其中C为通道数，或类别数

我的类别为10类，其中的值为1~10，所以只需要减1即可，如下图所示。

c_loss = nn.CrossEntropyLoss()
labels_v = labels_v-1 
loss0 = c_loss(d0, labels_v.long())
1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/89174