my code is to segment sequence image ,the annotaiton is the ground truth, its shape is BatchframesHeightWidth. There are 4 objects to segment in each image, the outputs of predicted segmentation is BatchFramesChannelHeight*Width. By rights, the outputs maxed torch.argmax have 4 values , however, why are there only two values?