Spaces:
Runtime error
Runtime error
File size: 850 Bytes
56ef371 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from mmengine import Registry, build_from_cfg
BACKBONES = Registry("backbone")
POS_EMBEDDINGS = Registry("position_embedding")
FUSERS = Registry("fuser")
ENCODERS = Registry("encoder")
DECODERS = Registry("decoder")
ARCHITECTURES = Registry("architecture")
def build_backbone(cfg):
"""Build encoder."""
return build_from_cfg(cfg, BACKBONES)
def build_position_embedding(cfg):
"""Build position embedding."""
return build_from_cfg(cfg, POS_EMBEDDINGS)
def build_fuser(cfg):
"""Build fuser."""
return build_from_cfg(cfg, FUSERS)
def build_encoder(cfg):
"""Build encoder."""
return build_from_cfg(cfg, ENCODERS)
def build_decoder(cfg):
"""Build decoder."""
return build_from_cfg(cfg, DECODERS)
def build_architecture(cfg):
"""Build architecture."""
return build_from_cfg(cfg, ARCHITECTURES)
|