将结构相同,权值不同网络融合的办法
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())