出现RuntimeError: CUDA error: device-side assert triggered错误的时候看错误之前出错的代码大概可以追溯到错误源。笔者出错的错误源显示为torch.nn.CrossEntropyLoss
这一般是由于标签和模型的类别数不匹配导致的,需要修改一致。
RuntimeError: CUDA error: device-side assert triggered还有可能是out of memory等错误导致的。
出现RuntimeError: CUDA error: device-side assert triggered错误的时候看错误之前出错的代码大概可以追溯到错误源。笔者出错的错误源显示为torch.nn.CrossEntropyLoss
这一般是由于标签和模型的类别数不匹配导致的,需要修改一致。
RuntimeError: CUDA error: device-side assert triggered还有可能是out of memory等错误导致的。