Info-HCVAE icon indicating copy to clipboard operation
Info-HCVAE copied to clipboard

Published 20 hours ago • verified publisher icon seanie12

error while running Google collab GPU

Open MPS-MASTER opened this issue 3 years ago • 1 comments

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

MPS-MASTER avatar Oct 27 '21 18:10 MPS-MASTER

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

Please refer to the following links

seanie12 avatar Oct 28 '21 01:10 seanie12