Shortcuts

mmengine.hub.get_model

mmengine.hub.get_model(cfg_path, pretrained=False, **kwargs)[source]

Get built model from external package.

Parameters
  • cfg_path (str) – External relative config path with prefix ‘package::’ and without suffix.

  • pretrained (bool) – Whether to load pretrained model. Defaults to False.

  • kwargs (dict) – Default arguments to build model.

Examples

>>> model = get_model('mmdet::faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py', pretrained=True)
>>> type(model)
<class 'mmdet.models.detectors.faster_rcnn.FasterRCNN'>
Returns

Built model.

Return type

nn.Module

Parameters
  • cfg_path (str) –

  • pretrained (bool) –

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.