Oh that makes sense.
I made the changes, but I get this error:
RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same

How do I change my input type?