[ModuleNotFoundError: No module named 'mmcv.cnn.utils.weight_init']

[jinglin80]

Branch

main branch (mmpretrain version)

Describe the bug

Tried to use swin_transformer.py but got an import error for line 11 of swin_transformer.py

ModuleNotFoundError: No module named 'mmcv.cnn.utils.weight_init'

I used the docker build from Dockerfile.
May you help?

Environment

python -c "import mmpretrain.utils;import pprint;pprint.pp(dict(mmpretrain.utils.collect_env()))"
Output:
Traceback (most recent call last):
File "", line 1, in
AttributeError: module 'pprint' has no attribute 'pp'

Other information

Should we update the Dockerfile and/or swin_transformer.py?