RuntimeError: CUDA error: device-side assert triggered

RuntimeError: CUDA error: device-side assert triggered解决办法

这是由于数据标签默认是从0开始的,比如标签类别为5类,默认是[0, 1, 2, 3, 4],而自己是[1, 2, 3, 4, 5],只需要在标签数据读入的时候减去1就可以。RuntimeError: CUDA error: device-side assert triggered_第1张图片

你可能感兴趣的:(python)