Seems like the main problem is when the torch.tensor is converted to float. When it’s removed at least the  Run-time error is gone(appears instead when tensor is converted to .cuda()) but now how do I change to torch.float dtype?(already tried .type(torch.float))