網絡融合

將結構相同,權值不同網絡融合的辦法

    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())

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章