pytorch 的交叉熵损失总是报错，同样的结构改成MSELoss就没有问题，谁能告诉我怎么回事呀
loss_function = torch.nn.CrossEntropyLoss()
TypeError: FloatClassNLLCriterion_updateOutput received an invalid combination of arguments
这是报错截图
1个回答
不知道你整个代码，但是也许是你的目标值的tensor type的问题吧
你试试把你的tensor转成longtensor的类型，比如
my_target = Variable(torch.LongTensor([1]))
曾经是科比
2017-11-24 01:09
没错是Tensor和LongTensor的问题，谢谢回答，已经解决。
-
大鱼
2017-11-27 12:40
相关讨论
随便看看