Function remove_prefix_from_state_dict

PATH/core/utils.py:132–136

(state_dict, prefix)

def remove_prefix_from_state_dict(state_dict, prefix):
    for old_key in list(state_dict.keys()):
        if old_key.startswith(prefix):
            new_key = remove_prefix_string(old_key, prefix)
            state_dict[new_key] = state_dict.pop(old_key)


def load_state(path, model, ignore=[], optimizer=None, cuda=False, recover=False,