今天在使用pytorch的使用跑着跑着出现了这个问题:
1.原因:
使用了CrossEntropyLoss作为损失函数,因为Tensor默认是Float类型的,而使用CrossEntropyLos要用longtensor而不是floattensor。
2.解决方案:
一:将torch.nn.CrossEntropyLoss 换成是其他损失函数,例如torch.nn.MSELoss()
二:将对应的输出值转变为Long类型,例如y = Variable(torch.LongTensor(3).random_(5))