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

Read the Docs v: v0.7.0
Versions
latest
stable
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.