Shortcuts

PretrainedInit

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.

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.