[Pytorch] RuntimeError: Found dtype Long but expected Float

PyTorch Runtime Error: input type (double) and bias type (float) should be the same