Different type of weighting with NLLLoss2d

@richard s solution seems nice and clean. Some time ago I implemented it myself here. Maybe it’s still usefull. :wink:

2 Likes