code: def remove_module_load(state_dict):
      
      new_state_dict = OrderedDict()
      for k, v in state_dict.items(): new_state_dict[k[7:]] = v
      return new_state_dict
docs: create new OrderedDict that does not contain `module.`