PyTorch uses float32 by default to initialize the model’s parameters, tensors etc.
As you’ve described the input tensors are float64 tensors, which will create a dtype mismatch.
You would thus either have to transform the input data to float32 or the model to float64.