[Bug] Error occurred while obtaining the model parameters

[fengchezhendexiatian]

分支

main 分支 (mmpretrain 版本)

描述该错误

The provided command "python mmpretrain/tools/analysis_tools/get_flops.py mmpretrain/configs/hrnet/hrnet-w18_4xb32_in1k.py" resulted in the following error.

When providing the configuration file for another model structure, no error occurs.

环境信息

{'sys.platform': 'linux',
'Python': '3.8.18 (default, Sep 11 2023, 13:40:15) [GCC 11.2.0]',
'CUDA available': True,
'numpy_random_seed': 2147483648,
'GPU 0,1': 'NVIDIA GeForce RTX 3080 Ti',
'CUDA_HOME': '/usr',
'NVCC': 'Cuda compilation tools, release 9.1, V9.1.8',
'GCC': 'gcc (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0',
'PyTorch': '1.9.1+cu111',
'TorchVision': '0.10.1+cu111',
'OpenCV': '4.8.1',
'MMEngine': '0.9.0',
'MMCV': '2.0.1',
'MMPreTrain': '1.1.1+'}

其他信息

No response