I have another question related to my loss.
My batch size is 128, and my target and model outputs are with size 128.
I want to optimize the model with all 128 entries, not their mean or sum values.
How can I do that?
I used reduction=none but I dont know how to apply this loss