PretrainedInit

class mmengine.model.PretrainedInit(checkpoint, prefix=None, map_location='cpu')

Initialize module by loading a pretrained model.

Parameters:

• checkpoint (str) – the checkpoint file of the pretrained model should be load.

• prefix (str, optional) – the prefix of a sub-module in the pretrained model. it is for loading a part of the pretrained model to initialize. For example, if we would like to only load the backbone of a detector model, we can set prefix='backbone.'. Defaults to None.

• map_location (str) – map tensors into proper locations. Defaults to cpu.