类型错误: cross_entropy_loss():参数"input"(位置 1)必须是张量,而不是元组



我得到这个错误>>TypeError: cross_entropy_loss():参数'input'(位置1)必须是张量,而不是元组

我尝试了不同的方法将元组转换成张量,但是我无法解决这个问题。有人能帮我一下吗?

这是包含主args_setting的配置代码。

输入图片描述

这是训练代码

输入图片描述

这是NN

的模型输入图片描述

最后这是错误

输入图片描述

提前谢谢你。

如果您希望这些class_weights的数字是一个张量,您应该简单地使用

class_weights = torch.tensor([0.02, 1.02])

我建议你看看这个链接,了解如何创建张量

相关内容

最新更新