liuadan

problem

7

After dividing the pixels of the binary image by 255 and setting num_class to 2 for training, I encountered an error when loading the model for segmentation：Missing key(s) in state_dict:...