“Assertion `index >= -sizes[i] && index < sizes[i] && “index out of bounds“` failed.” 解决方法

最近在使用Pytorch训练时报错,出现了以下信息:

Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds
device-side assert triggered

可以判断时数组下标越界导致的,但是由于把数据加载到cuda里面,导致报错信息并不能帮助我们准确地定位错误位置,这时候,只需要将*.cuda()或者.*to(device)去掉,将模型放在cpu训练,就能看到准确的错误信息:

“Assertion `index >= -sizes[i] && index < sizes[i] && “index out of bounds“` failed.” 解决方法_第1张图片

你可能感兴趣的:(Pytorch,pytorch,深度学习,人工智能)