最近在使用Pytorch训练时报错,出现了以下信息:
Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds
device-side assert triggered
可以判断时数组下标越界导致的,但是由于把数据加载到cuda
里面,导致报错信息并不能帮助我们准确地定位错误位置,这时候,只需要将*.cuda()
或者.*to(device)
去掉,将模型放在cpu训练,就能看到准确的错误信息: