error while running Google collab GPU

[MPS-MASTER]

Runtime error : lengths should be 1-D CPU int64 tensor, got cuda:0 Long tensor

[seanie12]

Try lengths = lengths.cpu() even if you train the model in gpu.

Please refer to the following links