错误:在GPU 下训练网络出现这样的错误,step=45前运行的很顺利,但是一到step=46就冒出这样的错误。一开始以为是内存不足,但是减少batch_size后并不能运行。
尝试:有博客评论讲在CPU下运行错误就明显报出来了。于是出现下述错误:
# pytorch----Target 200 is out of bounds
这就说明了有样本的标签超出了样本类别数,后来仔细想想发现,一共200个样本,标记为1~200,可能是此处的错误。更改为0~199后就没有问题了。
#或者可能是网络的输出个数小于样本类别数。
另外:
为什么程序总是在step=46时出现错误,毕竟样本是随即选择的阿?
因为设置了pytorch以及numpy、cudnn等的seed,每次都能保证从开始运行时随机选择的都一样,因此每次都在step=46时出现了标签为200的样本,看来设置seed还是很重要的勒!
【pytorch】【解决】【GPU下】RuntimeError: CUDA error: device-side assert triggered
最新推荐文章于 2024-07-08 08:30:00 发布