xdecoder/Instruct-X-Decoder
163
1from .registry import model_entrypoints2from .registry import is_model3 4from .transformer_encoder_fpn import *5 6def build_encoder(config, *args, **kwargs):7 model_name = config['MODEL']['ENCODER']['NAME']8 9 if not is_model(model_name):10 raise ValueError(f'Unkown model: {model_name}')11 12 return model_entrypoints(model_name)(config, *args, **kwargs)