Function build_network

ram/archs/__init__.py:18–25

(opt,is_print=True)

def build_network(opt,is_print=True):
    opt = deepcopy(opt)
    network_type = opt.pop('type')
    net = ARCH_REGISTRY.get(network_type)(**opt)
    logger = get_root_logger()
    if is_print:
        logger.info(f'Network [{net.__class__.__name__}] is created.')
    return net