mmengine.registry.build_model_from_cfg

mmengine.registry.build_model_from_cfg(cfg, registry, default_args=None)

Build a PyTorch model from config dict(s). Different from build_from_cfg, if cfg is a list, a nn.Sequential will be built.

Parameters

• cfg (dict, list[dict]) – The config of modules, which is either a config dict or a list of config dicts. If cfg is a list, the built modules will be wrapped with nn.Sequential.

• registry (Registry) – A registry the module belongs to.

• default_args (dict, optional) – Default arguments to build the module. Defaults to None.

Returns

A built nn.Module.

Return type

nn.Module