我得到这个错误>>TypeError: cross_entropy_loss():参数'input'(位置1)必须是张量,而不是元组
我尝试了不同的方法将元组转换成张量,但是我无法解决这个问题。有人能帮我一下吗?
这是包含主args_setting的配置代码。
输入图片描述
这是训练代码
输入图片描述
这是NN
的模型输入图片描述
最后这是错误
输入图片描述
提前谢谢你。
如果您希望这些class_weights的数字是一个张量,您应该简单地使用
class_weights = torch.tensor([0.02, 1.02])
我建议你看看这个链接,了解如何创建张量