网络融合

Deadly 提交于 2020-02-18 18:55:24

将结构相同,权值不同网络融合的办法

    opt = parser.parse_args()
    UPSCALE_FACTOR = opt.upscale_factor
    CROP_SIZE = opt.crop_size
    NUM_EPOCHS = opt.num_epochs
    CUDA_NUM=opt.cuda_num
    learning_rate=opt.lr
    ref_img_dir=opt.ref_img
    Up_netG = Up_Generator(UPSCALE_FACTOR)
    # Up_netG.cpu().state_dict()['block2.bn1.running_var']+=2
    # print(Up_netG.cpu().state_dict()['block2.bn1.running_var'])
    # print(Up_netG.cpu().state_dict())
    for k,v in Up_netG.cpu().state_dict().items():
        Up_netG.cpu().state_dict()[k]+=10000
    print(Up_netG.cpu().state_dict())

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!