Hi guys, I am encountering a similar issue:
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight’
I made sure that my model is on gpu by invoking model  = model.cuda() and I still get the above error message. Any hint as of why this is happening is greatly appreciated.
Best