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)

模型

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/124065075