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解决办法
这是由于数据标签默认是从0开始的,比如标签类别为5类,默认是[0, 1, 2, 3, 4],而自己是[1, 2, 3, 4, 5],只需要在标签数据读入的时候减去1就可以。