Shortcuts

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.

参数
  • 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.

返回

A built nn.Module.

返回类型

nn.Module

Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.