11 lines
336 B
Python
Executable File
11 lines
336 B
Python
Executable File
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}')
|
|
|
|
return model_entrypoints(model_name)(config, **kwargs) |