def upgrade_state_dict_named(self, state_dict, name):
    super().upgrade_state_dict_named(state_dict, name)
    return state_dict
def upgrade_state_dict_named(self, state_dict, name):
    super().upgrade_state_dict_named(state_dict, name)
    return state_dict