Ditlev_Jorgensen:

RuntimeError: Expected object of type torch.DoubleTensor but found type torch.FloatTensor for argument #2 'weight'



Probably,  you may need to call .float() on your data and models to solve this.