Solved by KFrank in post #2 
              
              
                Hello Yuerno - 

Conventional wisdom and common practice suggest that you 
should use CrossEntropyLoss for a multi-class classification 
problem.  Reading between the lines, I understand your 
segmentation problem to be a straightforward (multi-label) 
classification problem, so I would think that C…