Thanks you. But I’m not understand when we deal with 4 classes (background + 3 classes),the output of model should be [batch_size, 4, h, w]，I what to know  1) if the target should use the onehot encode and 2) what the target should like ? ([batch_size, height, width] using onehot encode or no use)