Skip to content

Commit 71c901f

Browse files
Update sequence-to-one.lua
1 parent 9e59b8e commit 71c901f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/sequence-to-one.lua

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ rnn = nn.Sequential()
2121
:add(nn.SplitTable(1,2))
2222
:add(nn.Sequencer(r))
2323
:add(nn.SelectTable(-1)) -- this selects the last time-step of the rnn output sequence
24-
:add(nn.Linear(hiddenSize, nIndex))
24+
:add(nn.Linear(hiddenSize, nClass))
2525
:add(nn.LogSoftMax())
2626

2727
-- build criterion

0 commit comments

Comments
 (0)