带rnn cuda()的LSTM?
我有以下型号:带rnn cuda()的LSTM?,cuda,lstm,Cuda,Lstm,我有以下型号: model = nn.Sequential() model:add(nn.Sequencer(nn.LookupTable(nIndex, hiddenSize))) model:add(nn.Sequencer(nn.FastLSTM(hiddenSize, hiddenSize, rho))) model:add(nn.Sequencer(nn.Linear(hiddenSize, nIndex))) model:add(nn.Sequencer(nn.LogSoftMax(
model = nn.Sequential()
model:add(nn.Sequencer(nn.LookupTable(nIndex, hiddenSize)))
model:add(nn.Sequencer(nn.FastLSTM(hiddenSize, hiddenSize, rho)))
model:add(nn.Sequencer(nn.Linear(hiddenSize, nIndex)))
model:add(nn.Sequencer(nn.LogSoftMax()))
然后,我通过以下方式将模型放到cuda上:
model:cuda()
我尝试转发一个输入(cudatensor),结果它中断了
FastLSTM与cuda不兼容吗
信息:
[string "local f = function() return targets:cuda() en..."]:1: attempt to call method 'cuda' (a nil value)
我设法在cuda上引入了一些计算,并做了以下更改: -首先将模型和标准放在cuda上:
model=model:cuda()
criterion=criterion:cuda()
-其次,我建立了一个cuda张量表,我将其作为目标提供:
local targetscudatable={}
for i = 1, #targets do
table.insert(targetscudatable, targets[i]:cuda())
end
然后它就可以工作了,但我想知道是否可以像输入一样向cuda发送更多的数据。无论如何,我已经有了一个500%的提速,这还不错你忘了要cunn套装:
需要“cunn”