pytorch中出现错误

1. CuDNN error: CUDNN_STATUS_EXECUTION_FAILED

• loss.backward()时出现RuntimeError: CuDNN error: CUDNN_STATUS_EXECUTION_FAILED，因为类别标签超出范围，需要检查类别标签。