RuntimeError: “nll

Traceback (most recent call last):File "E:/relate_code/Gaitpart-master/main.py", line 149, in train_model()File "E:/relate_code/Gaitpart-master/main.py", line 98, in train_modelloss = criterion(outputs, labels)File "C:\Anaconda3\envs\gaitpart\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_implreturn forward_call(*input, **kwargs)File "C:\Anaconda3\envs\gaitpart\lib\site-packages\torch\nn\modules\loss.py", line 1163, in forwardreturn F.cross_entropy(input, target, weight=self.weight,File "C:\Anaconda3\envs\gaitpart\lib\site-packages\torch\nn\functional.py", line 2996, in cross_entropyreturn torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int' 【RuntimeError: “nll】loss计算时，数据类型出错，将：
loss = criterion(outputs, labels) 修改为：
loss = criterion(outputs, labels.long()) 即可解决！
参考链接：https://wanghao.blog.csdn.net/article/details/123503587?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1.pc_relevant_antiscanv2&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1.pc_relevant_antiscanv2&utm_relevant_index=2