Oh nice, so the problem is basically solved then if you do a
x = x.float().to(device)

?
Btw I think I was initially wrong when I said that it was a typo in the error message, I think it’s indeed correct regarding what’s called “argument #1” and “argument #2”, but yeah, it can be easily confusing