pytorch拷贝权重 模型迁移

目录

获取模型参数维度

模型迁移动态匹配

---

获取模型参数维度

    model = MobileFormer(**args)

    model_dict=model.state_dict()

    for to_k,to_v in model_dict.items():
        try:
            if (to_v.size()!=torch.Size([])):
                print(to_k,to_v.size(0))
        except Exception as e:
            print(e)

模型