Thank you! I have already tried also this kind of solution but nothing changed so…I decided to write my own training procedure, for this task, based on MSELoss and used as a predictor a Resnext50 with the Linear layer that gives the desired output.