RuntimeError: CUDA error: device-side assert triggered

RuntimeError: CUDA error: device-side assert triggered

该Error是在pytorch框架下出现的，其原因是样本的label种类大于了代码所设置的类别数。
建议将类别数调到与label种类数相等或略大于label类数。