zoukankan      html  css  js  c++  java
  • 修改模型参数名

    
    
    import torch
    import torch.nn as nn
    import torch.nn.functional as F
    from torchvision import models
    from model import TTSR
    from option import args
    # model = TTSR.TTSR(args)
    # print(model)
    model_path  ='./pretrain_model/model_00118.pt'
    ori_model_dict = torch.load(model_path)
    model_state_dict = {key.replace('MainNet.b_1.weight', 'MainNet.b_mod.1.weight'): ori_model_dict[key] for key in ori_model_dict }#将MainNet.b_1.weight修改为MainNet.b_mod.1.weight
    model_state_dict = {key.replace('MainNet.b_1.bias',  'MainNet.b_mod.1.bias'): model_state_dict[key] for key in model_state_dict }
    model_state_dict = {key.replace('MainNet.b_4.weight', 'MainNet.b_mod.4.weight'): model_state_dict[key] for key in model_state_dict }
    model_state_dict = {key.replace('MainNet.b_4.bias', 'MainNet.b_mod.4.bias'): model_state_dict[key] for key in model_state_dict }
    model_state_dict = {key.replace('MainNet.b_6.weight', 'MainNet.b_mod.6.weight'):model_state_dict[key] for key in model_state_dict }
    model_state_dict = {key.replace('MainNet.b_6.bias', 'MainNet.b_mod.6.bias'):model_state_dict[key] for key in model_state_dict }
    model_state_dict = {key.replace('MainNet.b_8.weight', 'MainNet.b_mod.8.weight'): model_state_dict[key] for key in  model_state_dict }
    model_state_dict = {key.replace('MainNet.b_8.bias', 'MainNet.b_mod.8.bias'): model_state_dict[key] for key in model_state_dict }
    torch.save(model_state_dict,'now.pt')
    
    #加载打印新的模型参数
    model_path_now  ='./now.pt'
    now_model_dict = torch.load(model_path_now)
    
    print(now_model_dict['MainNet.b_6.weight'])
    print(now_model_dict['MainNet.b_6.bias'])
    
    
    
    
    
  • 相关阅读:
    几个常用的url生成二维码的接口
    php 实现打印预览的功能
    php实现pdf导出和打印功能。
    PHP 写入缓存
    jQuery获取Select选择的Text和 Value(转)
    jQuery获取多种input值的方法
    php mysqli_get_server_version()函数
    大盘能涨多少点?
    让你懂起来
    楼市、股市后下一届ZF将用什么去做超发货币的蓄水池(
  • 原文地址:https://www.cnblogs.com/tingtin/p/14629516.html
Copyright © 2011-2022 走看看