CUDA ERROR: device-side assert triggered at 和unable to get repr for ＜class ‘torch.Tensor‘＞问题及解决思路

可能原因是你的label使用了插值函数order = 3

lb = scipy.ndimage.zoom(lb, (288 / 192, 288 / 192, 1), output=None, order=3, mode='constant')

导致label出现了小于0的值,或者大于类别数 - 1 的值(比如算上背景是0 1 2 三类,label出现了大于2和小于0的值),在计算loss的时候,就会出错

遇到这种问题,可以点进dataset里面

查看自己是否对label进行插值过,并且使用的是order等于3来进行插值

如果有,那么你也可以把order改为0试试,order改为0,插值完还是整数,就不会越界了

一般做完这个操作就解决了

还有

临时的解决办法,

把label大于类别数- 1 的值跟小于0的值截断掉,也可以解决问题

比如我的是5分类(加上背景),所以把大于4的值跟小于0的值截断

# lb[lb > 4] = 4
# lb[lb < 0] = 0

CUDA ERROR: device-side assert triggered at 和unable to get repr for ＜class ‘torch.Tensor‘＞问题及解决思路相关推荐

RuntimeError: cuda runtime error (59) : device-side assert triggered
问题在训练 Transformer 的过程中,pytorhc出现的问题:RuntimeError: cuda runtime error (59) : device-side assert trig ...
cuda runtime error (59) : device-side assert triggered when running transfer_learning_
cuda runtime error (59) : device-side assert triggered when running transfer_learning_ 参考:https://bl ...
pytorch runtime error(59):device-side assert triggered at XXX
跑模型的时候,遇到了这个问题,定位是在 char-embedding 中的 conv 层中,由于需要使用 pool1d, char embedding后的 size 为(N*seq_len, word ...
RuntimeError: CUDA error: device-side assert triggered和Unable to get repr for ＜class ‘torch.Tensor
项目场景: 一个深度学习模型,主要包括GCN与CNN两个部分问题描述跑代码的时候发现他一直出现RuntimeError: CUDA error: device-side assert trigge ...
PyTorch报错：RuntimeError: CUDA error: device-side assert triggered at /pytorch/aten/src/THC/generic
训练模型报错 RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/TH ...
Debug Pytorch: RuntimeError: CUDA error: device-side assert triggered
报错信息 RuntimeError: CUDA error: device-side assert triggered /pytorch/aten/src/THC/THCTensorScatterGa ...
已解决RuntimeError: CUDA error: device-side assert triggered异常的正确解决方法，亲测有效！！！
已解决RuntimeError: CUDA error: device-side assert triggered异常的正确解决方法,亲测有效!!! 文章目录报错问题解决方法福利报错问题粉丝 ...
解决：RuntimeError: CUDA error: device-side assert triggered
@[TOC]解决办法:RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be asynch ...
错误处理--CUDA error: device-side assert triggered（很有效）
embeding中词的总数设小了在函数 nn.Embedding(vocab_size, embedding_dimension) 中参数vocab_size 为你字典的总词数,如果在训练中有编号 ...

CUDA ERROR: device-side assert triggered at 和unable to get repr for ＜class ‘torch.Tensor‘＞问题及解决思路

CUDA ERROR: device-side assert triggered at 和unable to get repr for ＜class ‘torch.Tensor‘＞问题及解决思路相关推荐

最新文章

热门文章