Hi, can you please explain how you solved the problem? I have the same problem. Please have a look once Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_mm 188