PretrainedInit — mmengine 0.10.7 documentation (original) (raw)

class mmengine.model.PretrainedInit(checkpoint, prefix=None, map_location='cpu')[source]¶

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.