我有一个火炬代码,可以在一台机器上正常运行,但错误" THNN.lua:110:输入和gradOutput形状不匹配:输入[90 x 271],gradOutput [271]&# 34;在另一台机器上。数据相同。我认为是版本问题。但我不知道如何解决它。有什么帮助吗?
问题是指logsoftmax模块的后退。但是当我打印出形状时,没有问题。
tack traceback:
[C]: in function 'v'
/root/torch/install/share/lua/5.1/nn/THNN.lua:110: in function 'LogSoftMax_updateGradInput'
/root/torch/install/share/lua/5.1/nn/LogSoftMax.lua:12: in function 'updateGradInput'
/root/torch/install/share/lua/5.1/nngraph/gmodule.lua:420: in function 'neteval'
/root/torch/install/share/lua/5.1/nngraph/gmodule.lua:454: in function 'updateGradInput'