Hi, if I am going to use CrossEntropyLoss, what should be the setup of the output and target?
Currently, I have 5 classes. If I make the mask a 1 channel image with class numbers, should I put 0 as background? So I shall have 0, 1, 2, 3, 4, 5 so 6 classes?