Please include a complete code on how to load a “checkpoint.pth.tar” file and then use your code to change the key like you mentioned so we can  use model.load_state_dict(torch.load(PATH)) for more prediction
thanks