Pepi:

Could it be a problem with the data itself?


Could be the case, e.g. if the ranges are large and you don’t normalize the data.
I assume that you don’t expect mismatches in the data, which could break the training.
As said before, your model is quite “small” (single linear layer), which might not be suitable to fit the data.