pytorch 的交叉熵损失总是报错,同样的结构改成MSELoss就没有问题

  统计/机器学习 深度学习 Python 人工神经网络
0

pytorch 的交叉熵损失总是报错,同样的结构改成MSELoss就没有问题,谁能告诉我怎么回事呀

loss_function = torch.nn.CrossEntropyLoss()

这是报错截图


 

大鱼   2017-11-23 22:29



   1个回答 
2

不知道你整个代码,但是也许是你的目标值的tensor type的问题吧

你试试把你的tensor转成longtensor的类型,比如

my_target = Variable(torch.LongTensor([1]))


曾经是科比   2017-11-24 01:09

没错是Tensor和LongTensor的问题,谢谢回答,已经解决。 - 大鱼   2017-11-27 12:40


相关问题

神经网络中的gradient check是什么意思?   2回答

深度学习和人工神经网络是一回事吗?   1回答

多项式网络(polynomial networks)的结构和激活函数?   0回答

神经网络中的dense和sparse是什么意思?   2回答

训练神经网络中经常提到的epoch是什么意思   2回答

神经网络中的Dropout和Bagging   1回答

前馈神经网络如何选择隐藏层的数量   2回答

人工神经网络有哪些常用的激活函数?   2回答

deep learning中的pooling是什么意思?   3回答

用神经网络做二元分类,输出层用Sigmoid还是Softmax?   1回答

keras中怎么增加Leaky relu的激活层   0回答

LeakyReLu是什么意思?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!