CUDA RuntimeError问题

RuntimeError: CUDA error: device-side assert triggered

原因

pytorch在做多分类任务的时候出现类别数目的问题。
在自己写的model里有一个参数是类别的数目，在训练的时候自己手动设置。问题是，输入的时候手误，导致数据里真实的类别数目和手动输入的类别数目不匹配。

解决

核对类别数目，传入正确的类别数量。