栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch 加载不对齐预训练

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch 加载不对齐预训练

以前改变网络通道数,需要重新从头训练,无法加载预训练,今天研究了一下如何改变网络通道后,还有预训练模型可用,这样可以减少980%的训练时间,提供训练效率。

废话不说,直接上代码:

这个代码加载预训练模型后,再训练无效果:


    backbone = MobileFace_83_w(256,l_size=[2,6,8,4]).to(0)

    backbone_pth = os.path.join("/data/408800_net.pth")

    state_dict=torch.load(backbone_pth, map_location=torch.device(0))
    # backbone.load_state_dict(state_dict,strict=False)
    bone_dict=backbone.state_dict()

    # model_end=Model_end(256).to(0)

    new_state_dict = OrderedDict()
    for k, v in state_dict.items():
        head = k[:7]
        if head == 'module.':
            tmp_name = k[7:]  # remove `module.`
        else:
            tmp_name = k
            # continue
        need_v= bone_dict[tmp_name]

        if len(need_v.size())==1:
            if need_v.size(0)>v.size(0):
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/503541.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号