Surgically rename a variable within pytorch model.

rename_var.py

import torch

PATH = 'model.pth'

ckpt = torch.load(PATH)
# `ckpt` will likely be a nested dictionary with nested `state_dict`
ckpt['state_dict'] = {
  key.replace('old_name', 'new_name'): value
  for key, value in ckpt['state_dict'].items()
}
# You may need to be smarter if the names are less distinctive, e.g. include some prefix

torch.save(ckpt, PATH)
# You'd rather like to use a different name in case something went wrong