bad argument #1 to 'updateOutput' (2D或3D(批处理模式)张量预期)

我尝试使用torch已经有一个星期了,并且我在时序卷积方面遇到了问题。 我正在UCR数据集上工作(请参阅http://www.cs.ucr.edu/~eamonn/time_series_data/),当我尝试卷积时,我遇到了这个错误:

/home/arthur/torch/install/bin/luajit: /home/arthur/torch/install/share/lua/5.1/nn/Sequential.lua:44: bad argument #1 to 'updateOutput' (2D or 3D(batch mode) tensor expected)
stack traceback:
    [C]: in function 'updateOutput'
    /home/arthur/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
    testTimeSeries.lua:59: in main chunk
    [C]: in function 'dofile'
    ...thur/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk
    [C]: at 0x00406670

我在lua中有这个模型:

net = nn.Sequential()
net:add(nn.TemporalConvolution(5, 1, 3))
net:add(nn.ReLU())
net:add(nn.Linear(trainset:sizeData(), 150))
net:add(nn.ReLU())
net:add(nn.Linear(150, #classes))      -- #classes是数据中的类别数
net:add(nn.LogSoftMax())
net = net:cuda()

我使用:train()训练了我的模型(但手动进行,我遇到了相同的错误)。 我知道卷积是如何工作的,但我不知道如何解决这个问题。 我想我必须重新调整我的输入,但如何? 如果您能帮助我,我将不胜感激!

谢谢, Arthur

ps:对不起我的英语

点赞