RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘解决办法
最新推荐文章于 2024-04-18 07:45:00 发布
该篇文章讨论了在使用nll_loss_forward_reduce_cuda_kernel_2d_index时遇到的RuntimeError,原因在于该函数未实现对Int类型的处理。博主在代码中添加了将target类型转换为torch.int64以解决这个问题,关注点在于CUDA和深度学习库CUDNN的兼容性问题。
摘要由CSDN通过智能技术生成