Hey there, I recently updated my torch and rnn( including dpnn, torchx ) and after updating the maskzero layer does not seem to work. The following code demonstrates the problem:
require 'rnn'
require 'nn'
cuda = arg[1]
mdl = nn.Sequential()
mdl:add(nn.MaskZero(nn.Linear(10,20),1))
mdl:add(nn.ReLU())
mdl:add(nn.LogSoftMax())
inp = torch.rand(10,10)
if cuda == "cuda" then
require 'cunn'
mdl = mdl:cuda()
inp = inp:cuda()
end
print(mdl:forward(inp))
When I run this script on the cpu, it works without any problem, but if I use it on the GPU, it crashes with:
Hey there, I recently updated my torch and rnn( including dpnn, torchx ) and after updating the maskzero layer does not seem to work. The following code demonstrates the problem:
When I run this script on the cpu, it works without any problem, but if I use it on the GPU, it crashes with: