RuntimeError: CUDA error: device-side assert triggered
File "ddp/run.py", line 225, in train target =torch.from_numpy(images_train[i_test]).cuda() RuntimeError: CUDA error: device-side assert triggered
cu:92: operator(): block: [3825,0,0], thread: [31,0,0] Assertion
index >= -sizes[i] && index < sizes[i] && “index out of bounds” failed. train()
提示:这里报错是指数组越界了!!!!!!
但是报错不是这一行,而是上一行!
代码如下(示例):
test = rgbshdr[i_test]
target =torch.from_numpy(images_train[i_test]).cuda()