import torch
path_2='./model.pth'
model_2=dict(torch.load(path_2))
model_2['state_dict'] = model_2.pop('params')
dict=[]
for k in model_2['state_dict'].keys():
k_="{}".format(k)
dict.append(k)
for k in dict:
k_="{}".format(k)
older_val=model_2['state_dict'][k_]
k_new="generator.{}".format(k_)
model_2['state_dict'][k_new] = model_2['state_dict'].pop(k_)
torch.save(model_2,'./model_changed.pth')
print(model_2)
import torch
path_1='/model_1.pth'
path_2='/model_2.pth'
model_1=torch.load(path_1)
model_2=torch.load(path_2)
for k,v in model_2.items():
for i,j in v.items():
model_1['state_dict'][i]=j
torch.save(model_1,'./model_3.pth')