Index _ | A | B | C | D | E | F | G | I | L | M | N | O | P | Q | R | S | T | U | V | W _ __init__() (trove.containers.materialized_qrel.MaterializedQRel method) (trove.containers.result_heapq.ResultHeapq method) (trove.containers.result_heapq_fast.FastResultHeapq method) (trove.data.ir_dataset_binary.BinaryDataset method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) (trove.data.ir_encoding_dataset.EncodingDataset method) (trove.data.vector_cache_mixin.VectorCacheMixin method) (trove.evaluation.callback_wandb.WandbCallback method) (trove.evaluation.evaluator.RetrievalEvaluator method) (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) (trove.evaluation.metrics.IRMetrics method) (trove.evaluation.relevance_evaluator.RelevanceEvaluatorPlus method) (trove.modeling.encoder_contriever.ContrieverEncoder method) (trove.modeling.encoder_default.DefaultEncoder method) (trove.modeling.losses.InfoNCELoss method) (trove.modeling.pretrained_encoder.PretrainedEncoder method) (trove.modeling.pretrained_retriever.PretrainedRetriever method) A add_batch() (trove.evaluation.metrics.IRMetrics method) add_qrel_nested_dict() (trove.containers.result_heapq.ResultHeapq method) add_triplet() (trove.containers.result_heapq.ResultHeapq method) aggregate_results() (trove.evaluation.metrics.IRMetrics method) all_devices_are_similar() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) all_gather_object() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) all_rec_ids (trove.data.ir_encoding_dataset.EncodingDataset property) (trove.data.vector_cache_mixin.VectorCacheMixin property) allowed_on_this_process() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) append_eos (trove.data.collator.RetrievalCollator attribute) as_qrel_nested_dict() (trove.containers.result_heapq.ResultHeapq method) as_sorted_lists() (trove.containers.result_heapq.ResultHeapq method) attn_implementation (trove.modeling.model_args.ModelArguments attribute) available_loaders() (in module trove.data.file_reader) available_losses() (trove.modeling.loss_base.RetrievalLoss class method) B barrier() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) BiEncoderRetriever (class in trove.modeling.retriever_biencoder) BinaryDataset (class in trove.data.ir_dataset_binary) bnb_4bit_quant_type (trove.modeling.model_args.ModelArguments attribute) broadcast_obj() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) broadcast_output (trove.evaluation.evaluation_args.EvaluationArguments attribute) C cache_dir (trove.containers.materialized_qrel.MaterializedQRel property) cache_records() (trove.data.vector_cache_mixin.VectorCacheMixin method) can_wrap() (trove.modeling.encoder_contriever.ContrieverEncoder class method) (trove.modeling.encoder_default.DefaultEncoder class method) (trove.modeling.pretrained_encoder.PretrainedEncoder class method) check_training_arguments() (trove.evaluation.metrics.IRMetrics method) cleanup_temp_artifacts (trove.evaluation.evaluation_args.EvaluationArguments attribute) compute() (trove.evaluation.metrics.IRMetrics method) compute_loss() (trove.trainer.RetrievalTrainer method) compute_scores() (trove.modeling.pretrained_encoder.PretrainedEncoder method) ContrieverEncoder (class in trove.modeling.encoder_contriever) corpus_cache (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) corpus_path (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) create_group_filter_fn() (trove.containers.materialized_qrel.MaterializedQRel static method) create_group_from_cache() (trove.data.ir_dataset_binary.BinaryDataset method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) create_group_on_the_fly() (trove.data.ir_dataset_binary.BinaryDataset method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) create_score_transform() (trove.containers.materialized_qrel.MaterializedQRel static method) D data_args (trove.data.collator.RetrievalCollator attribute) DataArguments (class in trove.data.data_args) dataset_name (trove.data.data_args.DataArguments attribute) DefaultEncoder (class in trove.modeling.encoder_default) device (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin property) disable_cache() (trove.data.ir_encoding_dataset.EncodingDataset method) E effective_cache_file_name (trove.data.vector_cache_mixin.VectorCacheMixin property) encode() (trove.evaluation.evaluator.RetrievalEvaluator method) (trove.modeling.encoder_contriever.ContrieverEncoder method) (trove.modeling.encoder_default.DefaultEncoder method) (trove.modeling.pretrained_encoder.PretrainedEncoder method) encode_passage() (trove.modeling.pretrained_encoder.PretrainedEncoder method) encode_query() (trove.modeling.pretrained_encoder.PretrainedEncoder method) encoder_class (trove.modeling.model_args.ModelArguments attribute) encoding_cache_dir (trove.evaluation.evaluation_args.EvaluationArguments attribute) EncodingDataset (class in trove.data.ir_encoding_dataset) ensure_list_of_correct_dtype() (trove.containers.materialized_qrel_config.MaterializedQRelConfig method) epoch_and_seed() (trove.data.ir_dataset_binary.BinaryDataset method) evaluate() (trove.evaluation.evaluator.RetrievalEvaluator method) (trove.evaluation.relevance_evaluator.RelevanceEvaluatorPlus method) EvaluationArguments (class in trove.evaluation.evaluation_args) export_and_load_eval_cache() (trove.data.ir_dataset_multilevel.MultiLevelDataset method) export_and_load_train_cache() (trove.data.ir_dataset_binary.BinaryDataset method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) export_result_dump() (trove.containers.result_heapq.ResultHeapq method) (trove.containers.result_heapq_fast.FastResultHeapq method) F fair_sharding (trove.evaluation.evaluation_args.EvaluationArguments attribute) FastResultHeapq (class in trove.containers.result_heapq_fast) FileLoaderNotFoundError filter_fn (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) find_appropriate_wrapper() (trove.modeling.pretrained_encoder.PretrainedEncoder class method) fingerprint (trove.containers.materialized_qrel.MaterializedQRel property) (trove.data.ir_dataset_binary.BinaryDataset property) (trove.data.ir_dataset_multilevel.MultiLevelDataset property) flush() (trove.data.vector_cache_mixin.VectorCacheMixin method) format_passage() (trove.modeling.encoder_contriever.ContrieverEncoder method) (trove.modeling.encoder_default.DefaultEncoder method) (trove.modeling.pretrained_encoder.PretrainedEncoder method) format_query() (trove.modeling.encoder_contriever.ContrieverEncoder method) (trove.modeling.encoder_default.DefaultEncoder method) (trove.modeling.pretrained_encoder.PretrainedEncoder method) forward() (trove.modeling.loss_base.RetrievalLoss method) (trove.modeling.losses.InfoNCELoss method) (trove.modeling.pretrained_retriever.PretrainedRetriever method) (trove.modeling.retriever_biencoder.BiEncoderRetriever method) from_model_args() (trove.modeling.loss_base.RetrievalLoss class method) (trove.modeling.pretrained_encoder.PretrainedEncoder class method) (trove.modeling.pretrained_retriever.PretrainedRetriever class method) G gather_object() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) get_cached_value() (trove.data.vector_cache_mixin.VectorCacheMixin method) get_encoding_datasets() (trove.data.ir_dataset_multilevel.MultiLevelDataset method) get_free_port_on_master() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) get_global_qids() (trove.containers.materialized_qrel.MaterializedQRel method) get_qrel_nested_dict() (trove.containers.materialized_qrel.MaterializedQRel method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) get_related_recs() (trove.containers.materialized_qrel.MaterializedQRel method) get_related_recs_for_global_qid() (trove.containers.materialized_qrel.MaterializedQRel method) get_related_recs_for_local_qid() (trove.containers.materialized_qrel.MaterializedQRel method) get_shard_weights() (trove.evaluation.evaluator.RetrievalEvaluator method) get_shared_uuid() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) get_state_dict() (trove.containers.result_heapq.ResultHeapq method) gradient_checkpointing_enable() (trove.modeling.pretrained_encoder.PretrainedEncoder method) group_bottom_k (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) group_filter_fn (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) group_first_k (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) group_random_k (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) group_size (trove.data.data_args.DataArguments attribute) group_top_k (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) I ignore_cache() (trove.data.ir_encoding_dataset.EncodingDataset method) info (trove.containers.materialized_qrel.MaterializedQRel property) (trove.data.ir_dataset_binary.BinaryDataset property) (trove.data.ir_dataset_multilevel.MultiLevelDataset property) InfoNCELoss (class in trove.modeling.losses) init_infinite_barrier() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) initialize_trackers() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) ir_metrics_k_values (trove.evaluation.evaluation_args.EvaluationArguments attribute) ir_metrics_relevance_threshold (trove.evaluation.evaluation_args.EvaluationArguments attribute) IRMetrics (class in trove.evaluation.metrics) is_cache_available (trove.data.vector_cache_mixin.VectorCacheMixin property) is_distributed (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin property) is_local_main_process (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin property) is_main_process (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin property) L load_cache() (trove.data.vector_cache_mixin.VectorCacheMixin method) load_in_4bit (trove.modeling.model_args.ModelArguments attribute) load_in_8bit (trove.modeling.model_args.ModelArguments attribute) load_qids() (in module trove.data.file_reader) load_qrel() (in module trove.data.file_reader) load_records() (in module trove.data.file_reader) local_to_global_id() (trove.containers.materialized_qrel.MaterializedQRel method) log() (trove.evaluation.callback_wandb.WandbCallback method) log_metrics() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) logits (trove.modeling.pretrained_retriever.RetrieverOutput attribute) lora_alpha (trove.modeling.model_args.ModelArguments attribute) lora_dropout (trove.modeling.model_args.ModelArguments attribute) lora_modules_to_save (trove.modeling.model_args.ModelArguments attribute) lora_r (trove.modeling.model_args.ModelArguments attribute) lora_target_modules (trove.modeling.model_args.ModelArguments attribute) lora_task_type (trove.modeling.model_args.ModelArguments attribute) loss (trove.modeling.model_args.ModelArguments attribute) (trove.modeling.pretrained_retriever.RetrieverOutput attribute) M MaterializedQRel (class in trove.containers.materialized_qrel) MaterializedQRelConfig (class in trove.containers.materialized_qrel_config) max_score (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) merge_mined_qrels (trove.evaluation.evaluation_args.EvaluationArguments attribute) merge_result_dump() (trove.containers.result_heapq.ResultHeapq method) min_score (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) mine_hard_negatives() (trove.evaluation.evaluator.RetrievalEvaluator method) model_name_or_path (trove.modeling.model_args.ModelArguments attribute) model_revision (trove.modeling.model_args.ModelArguments attribute) ModelArguments (class in trove.modeling.model_args) module trove.data.file_reader trove.data.file_reader_functions MultiLevelDataset (class in trove.data.ir_dataset_multilevel) N nearest_neighbor_search() (trove.evaluation.evaluator.RetrievalEvaluator method) negative_passage_no_shuffle (trove.data.data_args.DataArguments attribute) no_annot_in_mined_hn (trove.evaluation.evaluation_args.EvaluationArguments attribute) normalize (trove.modeling.model_args.ModelArguments attribute) O open_cache_io_streams() (trove.data.vector_cache_mixin.VectorCacheMixin method) output_dir (trove.evaluation.evaluation_args.EvaluationArguments attribute) output_qrel_format (trove.evaluation.evaluation_args.EvaluationArguments attribute) P pad_to_multiple_of (trove.data.data_args.DataArguments attribute) passage (trove.modeling.pretrained_retriever.RetrieverOutput attribute) passage_max_len (trove.data.data_args.DataArguments attribute) passage_selection_strategy (trove.data.data_args.DataArguments attribute) pbar() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) pbar_mode (trove.evaluation.evaluation_args.EvaluationArguments attribute) per_device_matmul_batch_size (trove.evaluation.evaluation_args.EvaluationArguments attribute) pooling (trove.modeling.model_args.ModelArguments attribute) positive_passage_no_shuffle (trove.data.data_args.DataArguments attribute) precompute_corpus_embs (trove.evaluation.evaluation_args.EvaluationArguments attribute) prefer_cache() (trove.data.ir_encoding_dataset.EncodingDataset method) PretrainedEncoder (class in trove.modeling.pretrained_encoder) PretrainedRetriever (class in trove.modeling.pretrained_retriever) print_mode (trove.evaluation.evaluation_args.EvaluationArguments attribute) Q qids_from_queries_jsonl() (in module trove.data.file_reader_functions) qrel_from_csv() (in module trove.data.file_reader_functions) qrel_from_grouped_triplets() (in module trove.data.file_reader_functions) qrel_from_pickle() (in module trove.data.file_reader_functions) qrel_from_sydir_corpus() (in module trove.data.file_reader_functions) qrel_from_tevatron_training_data() (in module trove.data.file_reader_functions) qrel_path (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) query (trove.modeling.pretrained_retriever.RetrieverOutput attribute) query_cache (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) query_max_len (trove.data.data_args.DataArguments attribute) query_path (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) query_subset_path (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) R rank (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin property) register_loader() (in module trove.data.file_reader) RelevanceEvaluatorPlus (class in trove.evaluation.relevance_evaluator) remove_temp_cache_pardir() (trove.evaluation.evaluator.RetrievalEvaluator method) reset_state() (trove.containers.result_heapq.ResultHeapq method) (trove.containers.result_heapq_fast.FastResultHeapq method) (trove.data.vector_cache_mixin.VectorCacheMixin method) (trove.evaluation.metrics.IRMetrics method) ResultHeapq (class in trove.containers.result_heapq) RetrievalCollator (class in trove.data.collator) RetrievalEvaluator (class in trove.evaluation.evaluator) RetrievalEvaluatorUtilsMixin (class in trove.evaluation.evaluator_mixin_utils) RetrievalLoss (class in trove.modeling.loss_base) RetrievalTrainer (class in trove.trainer) RetrievalTrainingArguments (class in trove.trainer) RetrieverOutput (class in trove.modeling.pretrained_retriever) S save_eval_topk_logits (trove.evaluation.evaluation_args.EvaluationArguments attribute) save_pretrained() (trove.modeling.encoder_contriever.ContrieverEncoder method) (trove.modeling.encoder_default.DefaultEncoder method) (trove.modeling.pretrained_encoder.PretrainedEncoder method) (trove.modeling.pretrained_retriever.PretrainedRetriever method) score_transform (trove.containers.materialized_qrel_config.MaterializedQRelConfig attribute) search_topk (trove.evaluation.evaluation_args.EvaluationArguments attribute) set_index_lookup_storage_type() (trove.containers.materialized_qrel.MaterializedQRel method) (trove.data.ir_dataset_binary.BinaryDataset method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) set_trainer() (trove.data.ir_dataset_binary.BinaryDataset method) setup() (trove.evaluation.callback_wandb.WandbCallback method) shard() (trove.data.ir_encoding_dataset.EncodingDataset method) similarity_fn() (trove.modeling.pretrained_encoder.PretrainedEncoder method) T temperature (trove.modeling.model_args.ModelArguments attribute) temperature_learnable (trove.modeling.model_args.ModelArguments attribute) to_dict() (trove.containers.materialized_qrel_config.MaterializedQRelConfig method) (trove.data.data_args.DataArguments method) (trove.modeling.model_args.ModelArguments method) tokenizer (trove.data.collator.RetrievalCollator attribute) torch_dtype (trove.modeling.model_args.ModelArguments attribute) training_step() (trove.trainer.RetrievalTrainer method) trove.data.file_reader module trove.data.file_reader_functions module trove_logging_mode (trove.evaluation.evaluation_args.EvaluationArguments attribute) (trove.trainer.RetrievalTrainingArguments attribute) trust_remote_code (trove.modeling.model_args.ModelArguments attribute) U unload_cache() (trove.data.vector_cache_mixin.VectorCacheMixin method) unshard() (trove.data.ir_encoding_dataset.EncodingDataset method) update_cache_file_name() (trove.data.vector_cache_mixin.VectorCacheMixin method) update_cache_subdir() (trove.data.vector_cache_mixin.VectorCacheMixin method) update_distributed_state() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) update_file_name_to_id() (trove.containers.materialized_qrel.MaterializedQRel method) update_metadata() (trove.containers.materialized_qrel.MaterializedQRel method) (trove.data.ir_dataset_binary.BinaryDataset method) (trove.data.ir_dataset_multilevel.MultiLevelDataset method) update_topk_records() (trove.containers.result_heapq_fast.FastResultHeapq method) use_bnb_nested_quant (trove.modeling.model_args.ModelArguments attribute) use_peft (trove.modeling.model_args.ModelArguments attribute) use_rslora (trove.modeling.model_args.ModelArguments attribute) V VectorCacheMixin (class in trove.data.vector_cache_mixin) W WandbCallback (class in trove.evaluation.callback_wandb) world_size (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin property) write_json() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method) write_json_lines() (trove.evaluation.evaluator_mixin_utils.RetrievalEvaluatorUtilsMixin method)