pytorch拷贝权重 模型迁移

目录

获取模型参数维度

模型迁移动态匹配

模型拷贝权重封装：

获取模型参数维度

model = MobileFormer(**args)

    model_dict=model.state_dict()

    for to_k,to_v in model_dict.items():
        try: