Hi,
You might want to double check your code to make sure that all the Tensors and models are of the same dtype. In this case it looks like you expect float everywhere.
You should be especially careful if you create Tensors from numpy arrays as they are in double precision by default and you will have to explicitly change them to float.