from transformers import PretrainedConfig

class SegformerPlusPlusConfig(PretrainedConfig):
    model_type = "segformer_plus_plus"

    def __init__(self, backbone_cfg=None, decode_head_cfg=None, out_channels=19, **kwargs):
        super().__init__(**kwargs)
        self.backbone_cfg = backbone_cfg
        self.decode_head_cfg = decode_head_cfg
        self.out_channels = out_channels