mmengine.hub.get_model

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

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) –