predict word is out of vocabulary about mem2seq HOT 3 CLOSED

Hi,

From the error you have, it seems you are using pytorch 0.4. Our code has been developed using pytorch 0.3. Maybe that is the problem. Can you try to downgrade your pytorch version (https://pytorch.org/previous-versions/)?

Soon we will update our code to pytorch 0.4 :)

from mem2seq.

yeah...it's true. it works fine that add .item() after topvi.squeeze()[i].
Thank you again.

from mem2seq.

Great,

pls be also careful to add .item() everywhere. In particular in toppi, else it will compare tensor to an integer, and it will always return false (i.e. always getting from vocab).

So use something like:
int(toppi.squeeze()[i].item())

Best
Andrea

from mem2seq.