【发布时间】:2017-02-06 23:14:53
【问题描述】:
我正在尝试在 Torch 中实现 RNN。为了习惯它,我从一个简单的任务开始,即预测序列中的下一个项目。序列是 {1,2,3,4,5,6,7,8,9,10} 的子序列,具有随机选择的偏移量。
我想实现一个带有 LSTM 单元的隐藏层的网络架构。这就是为什么我使用 nn.SeqLSTM(inputsize, outputsize)
rho = 5 -- number of steps for bptt
hiddensize = 15
inputsize = 1
outputsize = 1
seqlen = 5
nIndex = 10
batchsize = 4
seqlstm = nn.SeqLSTM(inputsize, outputsize)
criterion = nn.SequencerCriterion(nn.ClassNLLCriterion())
outputs = seqlstm:forward(inputs) -- inputs is seqlen x batchsize x inputsize
err = criterion:forward(outputs, targets) -- targets is seqlen x batchsize x 1
- 我需要 nn.LookupTable 吗?
- 这段代码似乎有点简单,我猜它缺少一些胶水。但是缺少哪些部分才能使其完整?
【问题讨论】:
标签: torch recurrent-neural-network