File size: 388 Bytes
2483c4f |
1 2 3 4 5 6 7 8 9 10 11 |
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
|