def METHOD_NAME(state_dict):
    # NOTE: all cls states are used for prediction,
    #   we predict the index so omit all pretrained states for prediction.
    new_state_dict = {}
    for key in state_dict:
        value = state_dict[key]
        if key.startswith('cls'):
            # NOTE: all cls states are used for prediction,
            #   we predict the index so omit all pretrained states for prediction.
            continue
        new_state_dict[key] = value
    return new_state_dict