convert to tensor before to_device

Gopal2002 changed pull request status to merged

Sign up or log in to comment