报错如下:
RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Float'
一般来说这个问题是计算Loss时的报错。
解决方法:
将如下代码
loss_func(torch.squeeze(output), target)
改为
loss_func(torch.squeeze(output), target.long())
RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Float'
一般来说这个问题是计算Loss时的报错。
将如下代码
loss_func(torch.squeeze(output), target)
改为
loss_func(torch.squeeze(output), target.long())