NLLLoss2d Calculation

[mohitsharma916 (Mohit Sharma)]

When calculating NLLLoss2d, is the loss averaged or summed over pixels?

[richard]

By default, loss is averaged over pixels*batch_size. This behavior can be controlled: http://pytorch.org/docs/master/nn.html?highlight=nllloss#torch.nn.NLLLoss2d

[mohitsharma916 (Mohit Sharma)]

@mohitsharma916

calculating NLLLoss2d, is the loss averaged or summed over

Thanks. The documentation explains it very well.