Pytorch:RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

Question

I set my model and data to the same device,

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net.to(device)