GLEE/glee/backbone/build.py

from .registry import model_entrypoints
from .registry import is_model

from .backbone import *

def build_backbone(config, **kwargs):
    model_name = config['MODEL']['BACKBONE']['NAME']
    if not is_model(model_name):
        raise ValueError(f'Unkown model: {model_name}')
    model = model_entrypoints(model_name)(config, **kwargs)
    return model