训练模型时出现RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/TH
训练模型时出现如下错误:RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:29解决方案:优先检查一下是不是class_num和输入数据的类别数对不上,差一个数也不行。...
原创
2021-10-08 10:36:08 ·
1562 阅读 ·
0 评论