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

https://blog.youkuaiyun.com/a424466072/article/details/130761
最低0.47元/天 解锁文章

9247

被折叠的 条评论
为什么被折叠?



