We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9e59b8e commit 71c901fCopy full SHA for 71c901f
examples/sequence-to-one.lua
@@ -21,7 +21,7 @@ rnn = nn.Sequential()
21
:add(nn.SplitTable(1,2))
22
:add(nn.Sequencer(r))
23
:add(nn.SelectTable(-1)) -- this selects the last time-step of the rnn output sequence
24
- :add(nn.Linear(hiddenSize, nIndex))
+ :add(nn.Linear(hiddenSize, nClass))
25
:add(nn.LogSoftMax())
26
27
-- build criterion
0 commit comments