loss.backward()报错:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
可以检查一下是不是多分类标签不是0开始。
比如二分类的标签应该是0,1。而不是1,2或其他。
多分类RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
于 2024-04-20 17:40:08 首次发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
loss.backward()报错:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
可以检查一下是不是多分类标签不是0开始。
比如二分类的标签应该是0,1。而不是1,2或其他。