python ./tools/pruning/get_channel_units.py /home/hy/mmpretrain/configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py -c -i --output-path ./temp

[scuizhibin]

my environment:

mmcv 2.0.1
mmdet 3.3.0
mmengine 0.10.0
mmpretrain 1.0.0
mmrazor 1.0.0
model-index 0.1.11
modelindex 0.0.2
mpmath 1.3.0
msgpack 1.0.8
multidict 6.0.5
multiprocess 0.70.15
mypy-extensions 1.0.0
torch 2.0.1+cu118
torchaudio 2.0.2+cu118
torchvision 0.15.2+cu118

KeyError: 'ImageClassifier is not in the mmengine::model registry. Please check whether the value of ImageClassifier is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module'

[surpoloyang]

the same error