dict = torch.load('./back.pth') # 在该模型中,要寻找的key都在"net"参数下,所以要先用"dict['net']"来限定 for k, v in list(dict['net'].items()): # 如果不加一层list,修改动作会导致报错:【RuntimeError: OrderedDict mutated during iteration】 if k[0:3]=='fem': dict['net']['cff'+k[3:]]=dict['net'].pop(k) torch.save(dict, './save.pth')