RuntimeError: CUDA error: device-side assert triggered

在改代码时经常遇到这个错误,于是记录下来。

省流:网络输出类别数和数据集实际类别数不相等。

错误是在loss.backword()这一行发生的,但是在debug时发现在loss.backword()之前变量就已经出现错误了,如下图:
RuntimeError: CUDA error: device-side assert triggered_第1张图片

这是求完交叉熵损失的结果,代码如下:
RuntimeError: CUDA error: device-side assert triggered_第2张图片说明问提出在s_output和label_source上。查了查unable to get repr for class说是数据对不上,才发现s_output的输出维度和数据集的类别数对不上,于是改了输出类别数num_class。

你可能感兴趣的:(一堆bug,深度学习,人工智能)