File size: 238 Bytes
32b542e
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11

from uniperceiver.utils.registry import Registry

ENGINE_REGISTRY = Registry("ENGINE")
ENGINE_REGISTRY.__doc__ = """
Registry for engine
"""

def build_engine(cfg):
    engine = ENGINE_REGISTRY.get(cfg.ENGINE.NAME)(cfg)
    return engine