报错如下：

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())