Hello, I think I may have came across the same issue, I had longitutde and latitude and had different values when using numpy and using torch, the loss for a numpy approach of sklearn and a pytorch we’re pretty different due to the nature of my values on it. The issue has been solved by using

torch.tensor(values,dtype=torch.float64)

Instead of my initial

torch.Tensor(values)

Hope it helps