修改pytorch模型名称

最新推荐文章于 2025-02-16 23:29:49 发布

枫呱呱

最新推荐文章于 2025-02-16 23:29:49 发布

阅读量366

点赞数 3

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_35275007/article/details/140353186

pytorch 专栏收录该内容

3 篇文章

订阅专栏

    def rename_layers(self, module, prefix=''):
        for name, child in list(module.named_children()):
            # 如果需要添加前缀，则修改名称
            if prefix:
                new_name = prefix + name
                # 在这里重命名子模块
                setattr(module, new_name, child)
                # 删除旧的名称对应的子模块
                delattr(module, name)
    def rename_layers_back(self, module, prefix=''):
        for name, child in list(module.named_children()):
            # 如果需要添加前缀，则修改名称
            if prefix:
                new_name = name.replace(prefix, "")
                # 在这里重命名子模块
                setattr(module, new_name, child)
                # 删除旧的名称对应的子模块
                delattr(module, name)

                    # 更新当前网络的结构字典
                    model_dict.update(pretrained_dict)
                    model.load_state_dict(model_dict, strict=False)

strict = False # 部分参数名称不一致时可忽略，不会报错