pytorch修改model 参数名

state_dict = {k: v for k,v in state_dict.items() if k.startswith("module.xxx.")}
state_dict = {k.partition('module.xxx.')[2]: v for k, v in state_dict.items()}