import torch

ckp_path = './checkpoints/fashion_PATN/latest_net_netG.pth'
save_path = './checkpoints/fashion_PATN_v1.0/latest_net_netG.pth'
states_dict = torch.load(ckp_path)
states_dict_new = states_dict.copy()
for key in states_dict.keys():
	if "running_var" in key or "running_mean" in key:
		del states_dict_new[key]

torch.save(states_dict_new, save_path)
[((5, 14, 5, 34), 'torch.load', 'torch.load', ({(5, 25, 5, 33): 'ckp_path'}, {}), '(ckp_path)', False, 'import torch\n'), ((11, 0, 11, 38), 'torch.save', 'torch.save', ({(11, 11, 11, 26): 'states_dict_new', (11, 28, 11, 37): 'save_path'}, {}), '(states_dict_new, save_path)', False, 'import torch\n')]