The results with reduction='sum' and reduction='mean' are slightly the same (I had to adapt the way I was displaying the loss). The success rate still stuck at 10% (predicting only 5s)
The F.relu didn’t provide any significant improvement.
The images are 128x128, it’s just a dataset I made for testing.
What else can I try? How can I debug where is the problem? Honestly, I am running out of ideas.
Thanks again