Pytorch 0.4.0版本保存的模型在高版本调用问题的解决方式

module.norm1.norm_func.running_mean” and “module.norm1.norm_func.running_var” 
for InstanceNorm2d with track_running_stats=False. If state_dict is a checkpoint saved
 before 0.4.0, this may be expected because InstanceNorm2d does not track running stats
  by default since 0.4.0. Please remove these keys from state_dict. If the running stats
   are actually needed, instead set track_running_stats=True in InstanceNorm2d to enable 
   them. See the documentation of InstanceNorm2d for details.

model_dict = torch.load(args.test_weight_path)
model_dict_clone = model_dict.copy()
for key, value in model_dict_clone.items():
    if key.endswith(('running_mean', 'running_var')):
        del model_dict[key]

Gnet.load_state_dict(model_dict,False)

norm_layer = functools.partial(
            nn.InstanceNorm2d, affine=False, track_running_stats=True)