RuntimeError: expected scalar type Long but found Float

最新推荐文章于 2024-06-10 22:18:41 发布

qq_43508163

最新推荐文章于 2024-06-10 22:18:41 发布

阅读量520

点赞数 2

文章标签： python 深度学习开发语言

本文链接：https://blog.csdn.net/qq_43508163/article/details/129272134

版权

Traceback (most recent call last):
File "main1.py", line 261, in <module>
loss = criterion(outputs, labels.squeeze(1)) + criterion_dl(outputs, labels)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/loss.py", line 1150, in forward
return F.cross_entropy(input, target, weight=self.weight,
File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/functional.py", line 2846, in cross_entropy
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing)
RuntimeError: expected scalar type Long but found Float

在label后加.long()

loss = criterion(outputs, labels.squeeze(1).long()) + criterion_dl(outputs, labels.long())