Actually I wish to do it with regression. The numbers in labels are integers but they don’t have to. They are like house prices or age. But even with 100 neurons in final layer the problem is the same. As a loss function I use mse_loss for regression and cross_entropy for classification (but I would rather to do regression). Any ideas why that happens?