在运行jcad_mindspore训练代码时出现报错,0.blocks.0.attn.relative_bias.index的形状在网络中是(20736, 529),而在加载的参数字典中却是(144, 144) (mindspore_py38) PS D:\glass\MindSpore_ModelZoo-main\MindSpore_ModelZoo-main\jcad_mindspore> python -m run.train_1 --config configs\UACANet_SwinB-S...
用来获取自己模型中对应层的预训练模型中的参数 state_dict = {k:v for k,v in save_model.items() if k in model_dict.keys()} state_dict.pop('fc.weight') # 丢掉FC层,因为输出不同,或_fc.weight state_dict.pop