错误的论点#1到' updateOutput' (预计2D或3D(批处理模式)张量)

时间:2016-02-16 15:00:36

标签: lua torch conv-neural-network

我上周以来一直在尝试火炬,我的时间卷积有问题。 我正在处理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 is the number of class in the data
net:add(nn.LogSoftMax())
net = net:cuda()

我用:train()训练我的模型(但手动,我有同样的错误)。 我理解卷积是如何工作的,但我不知道如何解决这个问题。 我想我必须调整我的输入但是怎么样? 如果你能帮助我,我会很感激!!

谢谢, 亚瑟

ps:抱歉我的英文

0 个答案:

没有答案
相关问题