@K_Frank I see you have answered questions (example) related to BCELoss (which is where I think the problem is). Can you give some insights into my problem please?
Thank you for you time.