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: latest
Versions
latest
stable
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.