If you are using nn.CrossEntropyLoss or nn.NLLLoss as your criterion, the output of your model should have the shape [batch_size, nb_classes, height, width], while your target should be a LongTensor in the shape [batch_size, height, width] containing the class indices.
If you are not dealing with a background class, the number of classes should correspond to the number of organs.