torch-net-toolkit icon indicating copy to clipboard operation
torch-net-toolkit copied to clipboard

Problems with nn.ConcatTable

Open davidstutz opened this issue 7 years ago • 1 comments

Hi,

I have difficulties when using net-toolkit with a network that contains a nn.ConcatTable layer. It usually fails in craftGradNBias at

module.gradInput = torch.Tensor():typeAs(module.output)

I tried several workarounds. I guess the problem is that gradInput as well as output might be tables. So I got some workarounds working, i.e. I am able to save models. But when I load them again, I get illegal memory accesses.

Any idea?

Thanks, David

davidstutz avatar May 22 '18 13:05 davidstutz

Can you provide me the smallest net and code snippet which will break the package? I can try to play around a bit.

Atcold avatar May 23 '18 00:05 Atcold