model.load_state_dict({k.replace('fc.1','fc'):v for k,v in torch.load('checkpoint.pt').items()}) #用'fc'代替'fc.1'