src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTForSemanticSegmentation/__init__
class TFMobileViTForSemanticSegmentation:    def __init__(self, config: MobileViTConfig, **kwargs) -> None:
        super().__init__(config, **kwargs)

        self.num_labels = config.num_labels
        self.mobilevit = TFMobileViTMainLayer(config, expand_output=False, name="mobilevit")
        self.segmentation_head = TFMobileViTDeepLabV3(config, name="segmentation_head")