Shortcuts

mmengine.registry.build_model_from_cfg

mmengine.registry.build_model_from_cfg(cfg, registry, default_args=None)[source]

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