Hi,
I am wondering how the result would be calculated if I set reduction=mean in torch.nn.BCELoss when I use auto-encoder model.
Do I average the loss only for “batch” dimension or both “pixel” and “batch”?
If I average for both “pixel” and “batch”, is there no way to sum loss for all “pixel” and to average for “batch” dimension?
Thanks