GPU issue

[yassmine-lam]

Hi,

I tried to run the aen model on GPU but I got this error

RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

could someone help plz?

Thank u

[Z.Zhao]

modify it to "lengths.cpu()"

[yassmine-lam]

yes u are right
thank u

[tusharkgerg]

@yassmine-lam im facing the same issue. Where and how were you able to fix this?