出现RuntimeError: CUDA error: device-side assert triggered错误的时候看错误之前出错的代码大概可以追溯到错误源。笔者出错的错误源显示为torch.nn.CrossEntropyLoss

        这一般是由于标签和模型的类别数不匹配导致的,需要修改一致。

        RuntimeError: CUDA error: device-side assert triggered还有可能是out of memory等错误导致的。

Logo

一站式 AI 云服务平台

更多推荐