nqg
nqg copied to clipboard
ParaSentDecoder hidden dim is nil while using pretrained checkpoints
In ParaSentDecoder, h_dim
is not in self.args
, thus it is not saved in the checkpoint - because only self.args
is saved. Due to this, the shape of gradContextInput
(https://github.com/xinyadu/nqg/blob/master/paragraph/onmt/modules/ParaSentDecoder.lua#L327) is incorrect and creates bad argument
error on updating gradients using gradContextInput:add(gradInput[self.args.inputIndex.context])
(https://github.com/xinyadu/nqg/blob/master/paragraph/onmt/modules/ParaSentDecoder.lua#L354)