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