Closed julien-c closed 8 years ago
I have the same problem. Did you train this on CPU? I think this may be a problem of CPU/GPU converting, the author wrote the codes for GPU but didn't adjust well for CPU. Have a look at this link: sample.lua fails to run: error in function addmm()
Problem solved! Add this function to seq2seq.lua:
-- function Seq2Seq:double()
-- created by zhaopku to fix the problem of CPU mode
function Seq2Seq:double()
self.encoder:double()
self.decoder:double()
if self.criterion then
self.criterion:double()
end
end
Then in train.lua:
at line 161:
change
model:float()
to
model:double()
Thx @zhaopku !
Thanks guys! Would it make sense to merge this to the codebase?
Thanks! It works for me!
Thanks, also works for me. Makes sense to merge (why not?)
yes, working here too. +1 for merge
I'm moderately convinced that this is a
neuralconvo
issue per se, but in case it is:Any idea?