Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_mm

[harsha_g]

Then, I am assuming you also put your model on the GPU. Didn’t you?