RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Double'
时间: 2023-12-04 18:04:57 浏览: 139
double 计算过程出现的误差
5星 · 资源好评率100%
这个错误通常出现在使用 PyTorch 进行模型训练时,如果你的设备支持 CUDA 的话,可以尝试将数据和模型移动到 GPU 上进行训练。如果你已经在 GPU 上训练了模型,那么这个错误可能是由于数据类型的问题。建议检查一下你的数据类型是否正确,比如是否使用了双精度浮点数(Double)而不是单精度浮点数(Float)。如果你的模型使用了双精度浮点数,那么你需要将输入数据也转换为双精度浮点数。
阅读全文