[Bug] Spark预训练权重加载错误！

[WWork1]

分支

main 分支 (mmpretrain 版本)

描述该错误

我按照文档，用下列代码加载Spark预训练权重。

from mmpretrain import get_model
model = get_model('spark_sparse-resnet50_800e_in1k', pretrained=True)

代码报错，提示了以下错误。

downsample_raito = H // SparseHelper._cur_active.shape[-1]
AttributeError: 'NoneType' object has no attribute 'shape'

环境信息

python xxx.py

其他信息

No response