Index

_

__init__() (lightning_ir.base.class_factory.LightningIRClassFactory method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory method) (lightning_ir.base.class_factory.LightningIRModelClassFactory method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.base.model.BatchEncodingWrapper method) (lightning_ir.base.model.LightningIRModel method) (lightning_ir.base.model.LightningIROutput method) (lightning_ir.base.module.LightningIRModule method) (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.bi_encoder.model.BiEncoderEmbedding method) (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.cross_encoder.model.CrossEncoderModel method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.data.data.DocSample method) (lightning_ir.data.data.IndexBatch method) (lightning_ir.data.data.QuerySample method) (lightning_ir.data.data.RankBatch method) (lightning_ir.data.data.RankSample method) (lightning_ir.data.data.SearchBatch method) (lightning_ir.data.data.TrainBatch method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.data.dataset.DocDataset method) (lightning_ir.data.dataset.QueryDataset method) (lightning_ir.data.dataset.RunDataset method) (lightning_ir.data.dataset.Sampler method) (lightning_ir.data.dataset.TupleDataset method) (lightning_ir.data.ir_datasets_utils.ScoredDocTuple method) (lightning_ir.data.ir_datasets_utils.ScoredDocTuples method) (lightning_ir.loss.loss.ApproxLossFunction method) (lightning_ir.loss.loss.ApproxMRR method) (lightning_ir.loss.loss.ApproxNDCG method) (lightning_ir.loss.loss.ApproxRankMSE method) (lightning_ir.loss.loss.ConstantMarginMSE method) (lightning_ir.loss.loss.EmbeddingLossFunction method) (lightning_ir.loss.loss.FLOPSRegularization method) (lightning_ir.loss.loss.InBatchCrossEntropy method) (lightning_ir.loss.loss.InBatchLossFunction method) (lightning_ir.loss.loss.InfoNCE method) (lightning_ir.loss.loss.KLDivergence method) (lightning_ir.loss.loss.L1Regularization method) (lightning_ir.loss.loss.L2Regularization method) (lightning_ir.loss.loss.ListwiseLossFunction method) (lightning_ir.loss.loss.LossFunction method) (lightning_ir.loss.loss.MarginMSE method) (lightning_ir.loss.loss.NeuralLossFunction method) (lightning_ir.loss.loss.PairwiseLossFunction method) (lightning_ir.loss.loss.RankNet method) (lightning_ir.loss.loss.RegularizationLossFunction method) (lightning_ir.loss.loss.ScoreBasedInBatchCrossEntropy method) (lightning_ir.loss.loss.ScoreBasedInBatchLossFunction method) (lightning_ir.loss.loss.ScoringLossFunction method) (lightning_ir.loss.loss.SupervisedMarginMSE method) (lightning_ir.main.LightningIRCLI method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.main.LightningIRTrainer method) (lightning_ir.main.LightningIRWandbLogger method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.t5.model.T5CrossEncoderModel method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) (lightning_ir.models.xtr.config.XTRConfig method) (lightning_ir.models.xtr.model.XTRModel method) (lightning_ir.models.xtr.model.XTRScoringFunction method) (lightning_ir.retrieve.base.indexer.IndexConfig method) (lightning_ir.retrieve.base.indexer.Indexer method) (lightning_ir.retrieve.base.searcher.SearchConfig method) (lightning_ir.retrieve.base.searcher.Searcher method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissFlatIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFPQIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissIVFPQIndexer method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexConfig method) (lightning_ir.retrieve.faiss.faiss_indexer.FaissPQIndexer method) (lightning_ir.retrieve.faiss.faiss_searcher.FaissSearchConfig method) (lightning_ir.retrieve.faiss.faiss_searcher.FaissSearcher method) (lightning_ir.retrieve.plaid.packed_tensor.PackedTensor method) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexConfig method) (lightning_ir.retrieve.plaid.plaid_indexer.PlaidIndexer method) (lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearchConfig method) (lightning_ir.retrieve.plaid.plaid_searcher.PlaidSearcher method) (lightning_ir.retrieve.plaid.residual_codec.ResidualCodec method) (lightning_ir.retrieve.sparse.sparse_indexer.SparseIndexConfig method) (lightning_ir.retrieve.sparse.sparse_indexer.SparseIndexer method) (lightning_ir.retrieve.sparse.sparse_searcher.SparseIndex method) (lightning_ir.retrieve.sparse.sparse_searcher.SparseSearchConfig method) (lightning_ir.retrieve.sparse.sparse_searcher.SparseSearcher method) (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.ConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) (lightning_ir.schedulers.schedulers.LambdaWarmupScheduler method) (lightning_ir.schedulers.schedulers.LinearSchedulerWithLinearWarmup method)

A

add_core_arguments_to_parser() (lightning_ir.main.LightningIRCLI method) add_default_arguments_to_parser() (lightning_ir.main.LightningIRCLI method) add_module() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) add_special_tokens() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) add_tokens() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) ADDED_ARGS (lightning_ir.base.config.LightningIRConfig attribute) (lightning_ir.bi_encoder.config.BiEncoderConfig attribute) (lightning_ir.cross_encoder.config.CrossEncoderConfig attribute) after_instantiate_classes() (lightning_ir.main.LightningIRCLI method) all_gather() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) all_special_ids (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) all_special_tokens (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property)

all_special_tokens_extended (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) ALLOW_SUB_BATCHING (lightning_ir.base.model.LightningIRModel attribute) apply() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) apply_chat_template() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) ApproxLossFunction (class in lightning_ir.loss.loss) ApproxMRR (class in lightning_ir.loss.loss) ApproxNDCG (class in lightning_ir.loss.loss) ApproxRankMSE (class in lightning_ir.loss.loss) as_target_tokenizer() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) automatic_optimization (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property)

B

backbone_model_type (lightning_ir.base.config.LightningIRConfig attribute) backward() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) batch_decode() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) batch_encode_plus() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) batch_encoding_wrapper() (in module lightning_ir.base.model) BatchEncodingWrapper (class in lightning_ir.base.model) before_instantiate_classes() (lightning_ir.main.LightningIRCLI method) bfloat16() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

BiEncoderConfig (class in lightning_ir.bi_encoder.config) BiEncoderEmbedding (class in lightning_ir.bi_encoder.model) BiEncoderModel (class in lightning_ir.bi_encoder.model) BiEncoderModule (class in lightning_ir.bi_encoder.module) BiEncoderOutput (class in lightning_ir.bi_encoder.model) BiEncoderTokenizer (class in lightning_ir.bi_encoder.tokenizer) buffers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) build_inputs_with_special_tokens() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method)

C

callback_metrics (lightning_ir.main.LightningIRTrainer property) cc_lir_model_type (lightning_ir.base.class_factory.LightningIRClassFactory property) (lightning_ir.base.class_factory.LightningIRConfigClassFactory property) (lightning_ir.base.class_factory.LightningIRModelClassFactory property) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory property) checkpoint_callback (lightning_ir.main.LightningIRTrainer property) checkpoint_callbacks (lightning_ir.main.LightningIRTrainer property) CHECKPOINT_MAPPING (in module lightning_ir.base.external_model_hub) children() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) ckpt_path (lightning_ir.main.LightningIRTrainer property) clean_up_tokenization() (lightning_ir.base.tokenizer.LightningIRTokenizer static method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer static method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer static method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer static method) clear() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) clip_gradients() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) ColConfig (class in lightning_ir.models.col) ColModel (class in lightning_ir.models.col) compile() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) config_class (lightning_ir.base.model.LightningIRModel attribute) (lightning_ir.base.tokenizer.LightningIRTokenizer attribute) (lightning_ir.bi_encoder.model.BiEncoderModel attribute) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer attribute) (lightning_ir.cross_encoder.model.CrossEncoderModel attribute) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer attribute) configure_callbacks() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) configure_gradient_clipping() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) configure_model() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method)

configure_optimizers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) configure_sharded_model() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) ConstantLRSchedulerWithLinearWarmup (class in lightning_ir.schedulers.lr_schedulers) ConstantMarginMSE (class in lightning_ir.loss.loss) ConstantSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) ConstantSchedulerWithQuadraticWarmup (class in lightning_ir.schedulers.schedulers) convert_tokens_to_string() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) copy() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) count() (lightning_ir.data.ir_datasets_utils.ScoredDocTuple method) cpu() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) create_qrels_from_dicts() (in module lightning_ir.base.validation_utils) create_run_from_scores() (in module lightning_ir.base.validation_utils) create_token_type_ids_from_sequences() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) CrossEncoderConfig (class in lightning_ir.cross_encoder.config) CrossEncoderModel (class in lightning_ir.cross_encoder.model) CrossEncoderModule (class in lightning_ir.cross_encoder.module) CrossEncoderOutput (class in lightning_ir.cross_encoder.model) CrossEncoderTokenizer (class in lightning_ir.cross_encoder.tokenizer) cuda() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) current_epoch (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) (lightning_ir.main.LightningIRTrainer property)

D

DASHED_DATASET_MAP (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) dataset (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) dataset_id (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) decode() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) default_root_dir (lightning_ir.main.LightningIRTrainer property) device (lightning_ir.bi_encoder.model.BiEncoderEmbedding property) device_ids (lightning_ir.main.LightningIRTrainer property) device_mesh (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) dict_torch_dtype_to_str() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method)

doc_embeddings (lightning_ir.bi_encoder.model.BiEncoderOutput attribute) doc_ids (lightning_ir.data.ir_datasets_utils.ScoredDocTuple attribute) DOC_TOKEN (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer attribute) doc_token_id (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) DocDataset (class in lightning_ir.data.dataset) docs (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) docs_dataset_id (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) DocSample (class in lightning_ir.data.data) double() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) download_artifact() (lightning_ir.main.LightningIRWandbLogger static method)

E

early_stopping_callback (lightning_ir.main.LightningIRTrainer property) early_stopping_callbacks (lightning_ir.main.LightningIRTrainer property) EmbeddingLossFunction (class in lightning_ir.loss.loss) embeddings (lightning_ir.bi_encoder.model.BiEncoderEmbedding attribute) (lightning_ir.cross_encoder.model.CrossEncoderOutput attribute) enable_validation (lightning_ir.main.LightningIRTrainer property) encode() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) (lightning_ir.models.xtr.model.XTRModel method) encode_doc() (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.xtr.model.XTRModel method) encode_plus() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) encode_query() (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.xtr.model.XTRModel method)

encoding (lightning_ir.bi_encoder.model.BiEncoderEmbedding attribute) estimated_stepping_batches (lightning_ir.main.LightningIRTrainer property) eval() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) evaluate_run() (in module lightning_ir.base.validation_utils) example_input_array (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) experiment (lightning_ir.main.LightningIRWandbLogger property) extra_repr() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

F

FaissFlatIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissFlatIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFPQIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissIVFPQIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissPQIndexConfig (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissPQIndexer (class in lightning_ir.retrieve.faiss.faiss_indexer) FaissSearchConfig (class in lightning_ir.retrieve.faiss.faiss_searcher) FaissSearcher (class in lightning_ir.retrieve.faiss.faiss_searcher) fit() (lightning_ir.main.LightningIRTrainer method) float() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) FLOPSRegularization (class in lightning_ir.loss.loss) forward() (lightning_ir.base.model.LightningIRModel method) (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.model.CrossEncoderModel method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.xtr.model.XTRModel method) (lightning_ir.models.xtr.model.XTRScoringFunction method) freeze() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) from_backbone_class() (lightning_ir.base.class_factory.LightningIRClassFactory method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory method) (lightning_ir.base.class_factory.LightningIRModelClassFactory method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) from_backbone_classes() (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method)

from_datasets() (lightning_ir.data.datamodule.LightningIRDataModule class method) from_dict() (lightning_ir.base.config.LightningIRConfig class method) (lightning_ir.bi_encoder.config.BiEncoderConfig class method) (lightning_ir.cross_encoder.config.CrossEncoderConfig class method) (lightning_ir.models.col.ColConfig class method) (lightning_ir.models.splade.SpladeConfig class method) (lightning_ir.models.t5.config.T5CrossEncoderConfig class method) (lightning_ir.models.xtr.config.XTRConfig class method) from_ir_dataset_sample() (lightning_ir.data.data.DocSample class method) (lightning_ir.data.data.QuerySample class method) from_json_file() (lightning_ir.base.config.LightningIRConfig class method) (lightning_ir.bi_encoder.config.BiEncoderConfig class method) (lightning_ir.cross_encoder.config.CrossEncoderConfig class method) (lightning_ir.models.col.ColConfig class method) (lightning_ir.models.splade.SpladeConfig class method) (lightning_ir.models.t5.config.T5CrossEncoderConfig class method) (lightning_ir.models.xtr.config.XTRConfig class method) from_pretrained() (lightning_ir.base.class_factory.LightningIRClassFactory method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory method) (lightning_ir.base.class_factory.LightningIRModelClassFactory method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory method) (lightning_ir.base.config.LightningIRConfig class method) (lightning_ir.base.model.LightningIRModel class method) (lightning_ir.base.tokenizer.LightningIRTokenizer class method) (lightning_ir.bi_encoder.config.BiEncoderConfig class method) (lightning_ir.bi_encoder.model.BiEncoderModel class method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer class method) (lightning_ir.cross_encoder.config.CrossEncoderConfig class method) (lightning_ir.cross_encoder.model.CrossEncoderModel class method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer class method) (lightning_ir.models.col.ColConfig class method) (lightning_ir.models.col.ColModel class method) (lightning_ir.models.splade.SpladeConfig class method) (lightning_ir.models.splade.SpladeModel class method) (lightning_ir.models.t5.config.T5CrossEncoderConfig class method) (lightning_ir.models.t5.model.T5CrossEncoderModel class method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer class method) (lightning_ir.models.xtr.config.XTRConfig class method) fromkeys() (lightning_ir.base.model.LightningIROutput class method) (lightning_ir.bi_encoder.model.BiEncoderOutput class method) (lightning_ir.cross_encoder.model.CrossEncoderOutput class method)

G

GenericConstantSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) GenericConstantSchedulerWithQuadraticWarmup (class in lightning_ir.schedulers.schedulers) GenericLinearSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) GenericScheduler (class in lightning_ir.schedulers.schedulers) get() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) get_backbone_config() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_backbone_model_type() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_buffer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) get_chat_template() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) get_config_dict() (lightning_ir.base.config.LightningIRConfig class method) (lightning_ir.bi_encoder.config.BiEncoderConfig class method) (lightning_ir.cross_encoder.config.CrossEncoderConfig class method) (lightning_ir.models.col.ColConfig class method) (lightning_ir.models.splade.SpladeConfig class method) (lightning_ir.models.t5.config.T5CrossEncoderConfig class method) (lightning_ir.models.xtr.config.XTRConfig class method) get_dataset_id() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) get_extra_state() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) get_last_lr() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) get_lightning_ir_config() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method)

get_lightning_ir_model_type() (lightning_ir.base.class_factory.LightningIRClassFactory static method) (lightning_ir.base.class_factory.LightningIRConfigClassFactory static method) (lightning_ir.base.class_factory.LightningIRModelClassFactory static method) (lightning_ir.base.class_factory.LightningIRTokenizerClassFactory static method) get_lr() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) get_output_embeddings() (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.xtr.model.XTRModel method) get_parameter() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) get_special_tokens_mask() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) get_submodule() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) get_text_config() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) get_vocab() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) global_rank (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) global_step (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) (lightning_ir.main.LightningIRTrainer property) group_separator (lightning_ir.main.LightningIRWandbLogger property)

H

half() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

hparams (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) (lightning_ir.data.datamodule.LightningIRDataModule property) hparams_initial (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) (lightning_ir.data.datamodule.LightningIRDataModule property)

I

InBatchCrossEntropy (class in lightning_ir.loss.loss) InBatchLossFunction (class in lightning_ir.loss.loss) index() (lightning_ir.data.ir_datasets_utils.ScoredDocTuple method) (lightning_ir.main.LightningIRTrainer method) IndexBatch (class in lightning_ir.data.data) IndexCallback (class in lightning_ir.callbacks.callbacks) IndexConfig (class in lightning_ir.retrieve.base.indexer) Indexer (class in lightning_ir.retrieve.base.indexer) inference_dataloader() (lightning_ir.data.datamodule.LightningIRDataModule method) InfoNCE (class in lightning_ir.loss.loss) init_module() (lightning_ir.main.LightningIRTrainer method) init_parser() (lightning_ir.main.LightningIRCLI method) instantiate_classes() (lightning_ir.main.LightningIRCLI method) instantiate_trainer() (lightning_ir.main.LightningIRCLI method)

ipu() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) ir_dataset (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) is_global_zero (lightning_ir.main.LightningIRTrainer property) is_last_batch (lightning_ir.main.LightningIRTrainer property) items() (lightning_ir.bi_encoder.model.BiEncoderEmbedding method)

K

KLDivergence (class in lightning_ir.loss.loss)

L

L1Regularization (class in lightning_ir.loss.loss) L2Regularization (class in lightning_ir.loss.loss) LambdaWarmupScheduler (class in lightning_ir.schedulers.schedulers) lightning_ir module lightning_ir.base module lightning_ir.base.class_factory module lightning_ir.base.config module lightning_ir.base.external_model_hub module lightning_ir.base.model module lightning_ir.base.module module lightning_ir.base.tokenizer module lightning_ir.base.validation_utils module lightning_ir.bi_encoder module lightning_ir.bi_encoder.config module lightning_ir.bi_encoder.model module lightning_ir.bi_encoder.module module lightning_ir.bi_encoder.tokenizer module lightning_ir.callbacks module lightning_ir.callbacks.callbacks module lightning_ir.cross_encoder module lightning_ir.cross_encoder.config module lightning_ir.cross_encoder.model module lightning_ir.cross_encoder.module module lightning_ir.cross_encoder.tokenizer module lightning_ir.data module lightning_ir.data.data module lightning_ir.data.datamodule module lightning_ir.data.dataset module lightning_ir.data.ir_datasets_utils module lightning_ir.loss module lightning_ir.loss.loss module lightning_ir.main module lightning_ir.models module lightning_ir.models.col module lightning_ir.models.splade module lightning_ir.models.t5 module lightning_ir.models.t5.config module lightning_ir.models.t5.model module lightning_ir.models.t5.tokenizer module lightning_ir.models.xtr module lightning_ir.models.xtr.config module lightning_ir.models.xtr.model module lightning_ir.retrieve module lightning_ir.retrieve.base module lightning_ir.retrieve.base.indexer module lightning_ir.retrieve.base.searcher module lightning_ir.retrieve.faiss module lightning_ir.retrieve.faiss.faiss_indexer module lightning_ir.retrieve.faiss.faiss_searcher module lightning_ir.retrieve.plaid module

lightning_ir.retrieve.plaid.packed_tensor module lightning_ir.retrieve.plaid.plaid_indexer module lightning_ir.retrieve.plaid.plaid_searcher module lightning_ir.retrieve.plaid.residual_codec module lightning_ir.retrieve.sparse module lightning_ir.retrieve.sparse.sparse_indexer module lightning_ir.retrieve.sparse.sparse_searcher module lightning_ir.schedulers module lightning_ir.schedulers.lr_schedulers module lightning_ir.schedulers.schedulers module LightningIRClassFactory (class in lightning_ir.base.class_factory) LightningIRCLI (class in lightning_ir.main) LightningIRConfig (class in lightning_ir.base.config) LightningIRConfigClassFactory (class in lightning_ir.base.class_factory) LightningIRDataModule (class in lightning_ir.data.datamodule) LightningIRModel (class in lightning_ir.base.model) LightningIRModelClassFactory (class in lightning_ir.base.class_factory) LightningIRModule (class in lightning_ir.base.module) LightningIROutput (class in lightning_ir.base.model) LightningIRSaveConfigCallback (class in lightning_ir.main) LightningIRTokenizer (class in lightning_ir.base.tokenizer) LightningIRTokenizerClassFactory (class in lightning_ir.base.class_factory) LightningIRTrainer (class in lightning_ir.main) LightningIRWandbLogger (class in lightning_ir.main) LinearLRSchedulerWithLinearWarmup (class in lightning_ir.schedulers.lr_schedulers) LinearSchedulerWithLinearWarmup (class in lightning_ir.schedulers.schedulers) link_optimizers_and_lr_schedulers() (lightning_ir.main.LightningIRCLI static method) ListwiseLossFunction (class in lightning_ir.loss.loss) load_from_checkpoint() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) load_state_dict() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) local_rank (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) log() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) log_audio() (lightning_ir.main.LightningIRWandbLogger method) log_dict() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) log_dir (lightning_ir.main.LightningIRTrainer property) (lightning_ir.main.LightningIRWandbLogger property) log_graph() (lightning_ir.main.LightningIRWandbLogger method) log_image() (lightning_ir.main.LightningIRWandbLogger method) log_random() (lightning_ir.data.dataset.Sampler static method) log_table() (lightning_ir.main.LightningIRWandbLogger method) log_text() (lightning_ir.main.LightningIRWandbLogger method) log_video() (lightning_ir.main.LightningIRWandbLogger method) logged_metrics (lightning_ir.main.LightningIRTrainer property) logger (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) (lightning_ir.main.LightningIRTrainer property) loggers (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) (lightning_ir.main.LightningIRTrainer property) LossFunction (class in lightning_ir.loss.loss) lr_scheduler_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) lr_schedulers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method)

M

main() (in module lightning_ir.main) manual_backward() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) MarginMSE (class in lightning_ir.loss.loss) max_len_sentences_pair (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) max_len_single_sentence (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) MLMHead (class in lightning_ir.bi_encoder.model) model (lightning_ir.main.LightningIRTrainer property) model_type (lightning_ir.base.config.LightningIRConfig attribute) (lightning_ir.bi_encoder.config.BiEncoderConfig attribute) (lightning_ir.cross_encoder.config.CrossEncoderConfig attribute) module lightning_ir lightning_ir.base lightning_ir.base.class_factory lightning_ir.base.config lightning_ir.base.external_model_hub lightning_ir.base.model lightning_ir.base.module lightning_ir.base.tokenizer lightning_ir.base.validation_utils lightning_ir.bi_encoder lightning_ir.bi_encoder.config lightning_ir.bi_encoder.model lightning_ir.bi_encoder.module lightning_ir.bi_encoder.tokenizer lightning_ir.callbacks lightning_ir.callbacks.callbacks lightning_ir.cross_encoder lightning_ir.cross_encoder.config lightning_ir.cross_encoder.model lightning_ir.cross_encoder.module lightning_ir.cross_encoder.tokenizer lightning_ir.data lightning_ir.data.data lightning_ir.data.datamodule lightning_ir.data.dataset lightning_ir.data.ir_datasets_utils lightning_ir.loss lightning_ir.loss.loss lightning_ir.main lightning_ir.models lightning_ir.models.col lightning_ir.models.splade lightning_ir.models.t5 lightning_ir.models.t5.config lightning_ir.models.t5.model lightning_ir.models.t5.tokenizer lightning_ir.models.xtr lightning_ir.models.xtr.config lightning_ir.models.xtr.model lightning_ir.retrieve lightning_ir.retrieve.base lightning_ir.retrieve.base.indexer lightning_ir.retrieve.base.searcher lightning_ir.retrieve.faiss lightning_ir.retrieve.faiss.faiss_indexer lightning_ir.retrieve.faiss.faiss_searcher lightning_ir.retrieve.plaid lightning_ir.retrieve.plaid.packed_tensor lightning_ir.retrieve.plaid.plaid_indexer lightning_ir.retrieve.plaid.plaid_searcher lightning_ir.retrieve.plaid.residual_codec lightning_ir.retrieve.sparse lightning_ir.retrieve.sparse.sparse_indexer lightning_ir.retrieve.sparse.sparse_searcher lightning_ir.schedulers lightning_ir.schedulers.lr_schedulers lightning_ir.schedulers.schedulers

modules() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) move_to_end() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) mtia() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

N

name (lightning_ir.main.LightningIRWandbLogger property) named_buffers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) named_children() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) named_modules() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

named_parameters() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) NeuralLossFunction (class in lightning_ir.loss.loss) num_devices (lightning_ir.main.LightningIRTrainer property) num_docs (lightning_ir.data.ir_datasets_utils.ScoredDocTuple attribute) num_labels (lightning_ir.base.config.LightningIRConfig property) (lightning_ir.bi_encoder.config.BiEncoderConfig property) (lightning_ir.cross_encoder.config.CrossEncoderConfig property) (lightning_ir.models.col.ColConfig property) (lightning_ir.models.splade.SpladeConfig property) (lightning_ir.models.t5.config.T5CrossEncoderConfig property) (lightning_ir.models.xtr.config.XTRConfig property) num_predict_batches (lightning_ir.main.LightningIRTrainer property) num_sanity_val_batches (lightning_ir.main.LightningIRTrainer property) num_test_batches (lightning_ir.main.LightningIRTrainer property) num_training_batches (lightning_ir.main.LightningIRTrainer property) num_val_batches (lightning_ir.main.LightningIRTrainer property)

O

on_after_backward() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_after_batch_transfer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) on_before_backward() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_before_batch_transfer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) on_before_optimizer_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_before_zero_grad() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_exception() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_fit_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_fit_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_gpu (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) on_load_checkpoint() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_predict_batch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_predict_batch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_predict_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_predict_epoch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_predict_epoch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_predict_model_eval() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) on_predict_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_sanity_check_end() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_sanity_check_start() (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_save_checkpoint() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_test_batch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method)

on_test_batch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_test_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_test_epoch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_test_epoch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_test_model_eval() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) on_test_model_train() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) on_test_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_batch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_batch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) on_train_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) on_train_epoch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_epoch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_train_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) on_validation_batch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_batch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_epoch_end() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_epoch_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) on_validation_model_eval() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) on_validation_model_train() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) on_validation_model_zero_grad() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) on_validation_start() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) optimizer_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) optimizer_zero_grad() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) optimizers() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method)

P

PackedTensor (class in lightning_ir.retrieve.plaid.packed_tensor) pad() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) pad_token_type_id (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) PairwiseLossFunction (class in lightning_ir.loss.loss) parameters() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) parse_arguments() (lightning_ir.main.LightningIRCLI method) PlaidIndexConfig (class in lightning_ir.retrieve.plaid.plaid_indexer) PlaidIndexer (class in lightning_ir.retrieve.plaid.plaid_indexer) PlaidSearchConfig (class in lightning_ir.retrieve.plaid.plaid_searcher) PlaidSearcher (class in lightning_ir.retrieve.plaid.plaid_searcher) popitem() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) POST_LOAD_CALLBACKS (in module lightning_ir.base.external_model_hub) predict() (lightning_ir.main.LightningIRTrainer method) predict_dataloader() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) predict_dataloaders (lightning_ir.main.LightningIRTrainer property) predict_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method)

prepare_data() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) prepare_for_model() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) prepare_input() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) prepare_seq2seq_batch() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) print() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRTrainer method) print_lr() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) progress_bar_callback (lightning_ir.main.LightningIRTrainer property) progress_bar_metrics (lightning_ir.main.LightningIRTrainer property) push_to_hub() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) (lightning_ir.models.xtr.config.XTRConfig method)

Q

qrels (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property) queries (lightning_ir.data.dataset.DocDataset property) (lightning_ir.data.dataset.QueryDataset property) (lightning_ir.data.dataset.RunDataset property) (lightning_ir.data.dataset.TupleDataset property)

query_embeddings (lightning_ir.bi_encoder.model.BiEncoderOutput attribute) query_id (lightning_ir.data.ir_datasets_utils.ScoredDocTuple attribute) QUERY_TOKEN (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer attribute) query_token_id (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) QueryDataset (class in lightning_ir.data.dataset) QuerySample (class in lightning_ir.data.data)

R

random() (lightning_ir.data.dataset.Sampler static method) RankBatch (class in lightning_ir.data.data) RankCallback (class in lightning_ir.callbacks.callbacks) RankNet (class in lightning_ir.loss.loss) RankSample (class in lightning_ir.data.data) re_rank() (lightning_ir.main.LightningIRTrainer method) received_sigterm (lightning_ir.main.LightningIRTrainer property) register_backward_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_buffer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_for_auto_class() (lightning_ir.base.config.LightningIRConfig class method) (lightning_ir.base.tokenizer.LightningIRTokenizer class method) (lightning_ir.bi_encoder.config.BiEncoderConfig class method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer class method) (lightning_ir.cross_encoder.config.CrossEncoderConfig class method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer class method) (lightning_ir.models.col.ColConfig class method) (lightning_ir.models.splade.SpladeConfig class method) (lightning_ir.models.t5.config.T5CrossEncoderConfig class method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer class method) (lightning_ir.models.xtr.config.XTRConfig class method) register_forward_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_forward_pre_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_full_backward_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_full_backward_pre_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

register_load_state_dict_post_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_load_state_dict_pre_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_module() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_msmarco() (in module lightning_ir.data.ir_datasets_utils) register_parameter() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_state_dict_post_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) register_state_dict_pre_hook() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) RegisterLocalDatasetCallback (class in lightning_ir.callbacks.callbacks) RegularizationLossFunction (class in lightning_ir.loss.loss) requires_grad_() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) ReRankCallback (class in lightning_ir.callbacks.callbacks) ResidualCodec (class in lightning_ir.retrieve.plaid.residual_codec) root_dir (lightning_ir.main.LightningIRWandbLogger property) RunDataset (class in lightning_ir.data.dataset)

S

sample() (lightning_ir.data.dataset.Sampler static method) Sampler (class in lightning_ir.data.dataset) sanitize_special_tokens() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) sanity_checking (lightning_ir.main.LightningIRTrainer property) save() (lightning_ir.main.LightningIRWandbLogger method) save_checkpoint() (lightning_ir.main.LightningIRTrainer method) save_config() (lightning_ir.main.LightningIRSaveConfigCallback method) save_dir (lightning_ir.main.LightningIRWandbLogger property) save_hyperparameters() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) save_pretrained() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.base.module.LightningIRModule method) (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) (lightning_ir.models.xtr.config.XTRConfig method) save_vocabulary() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) ScaleLinear (class in lightning_ir.models.t5.model) score() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.xtr.model.XTRModel method) ScoreBasedInBatchCrossEntropy (class in lightning_ir.loss.loss) ScoreBasedInBatchLossFunction (class in lightning_ir.loss.loss) ScoredDocTuple (class in lightning_ir.data.ir_datasets_utils) ScoredDocTuples (class in lightning_ir.data.ir_datasets_utils) scores (lightning_ir.data.ir_datasets_utils.ScoredDocTuple attribute) scoring_mask (lightning_ir.bi_encoder.model.BiEncoderEmbedding attribute) scoring_mask() (lightning_ir.bi_encoder.model.BiEncoderModel method) (lightning_ir.models.col.ColModel method) (lightning_ir.models.splade.SpladeModel method) (lightning_ir.models.xtr.model.XTRModel method) ScoringFunction (class in lightning_ir.bi_encoder.model) ScoringLossFunction (class in lightning_ir.loss.loss) search() (lightning_ir.main.LightningIRTrainer method) SearchBatch (class in lightning_ir.data.data) SearchCallback (class in lightning_ir.callbacks.callbacks) SearchConfig (class in lightning_ir.retrieve.base.searcher) Searcher (class in lightning_ir.retrieve.base.searcher) searcher (lightning_ir.bi_encoder.module.BiEncoderModule property) set_extra_state() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) set_optimizer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) set_submodule() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

setup() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) setup_parser() (lightning_ir.main.LightningIRCLI method) share_memory() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) single_relevant() (lightning_ir.data.dataset.Sampler static method) SparseIndex (class in lightning_ir.retrieve.sparse.sparse_searcher) SparseIndexConfig (class in lightning_ir.retrieve.sparse.sparse_indexer) SparseIndexer (class in lightning_ir.retrieve.sparse.sparse_indexer) SparseSearchConfig (class in lightning_ir.retrieve.sparse.sparse_searcher) SparseSearcher (class in lightning_ir.retrieve.sparse.sparse_searcher) special_tokens_map (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) special_tokens_map_extended (lightning_ir.base.tokenizer.LightningIRTokenizer property) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer property) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer property) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer property) SpladeConfig (class in lightning_ir.models.splade) SpladeModel (class in lightning_ir.models.splade) state_dict() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) STATE_DICT_KEY_MAPPING (in module lightning_ir.base.external_model_hub) state_key (lightning_ir.callbacks.callbacks.IndexCallback property) (lightning_ir.callbacks.callbacks.RankCallback property) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback property) (lightning_ir.callbacks.callbacks.ReRankCallback property) (lightning_ir.callbacks.callbacks.SearchCallback property) (lightning_ir.main.LightningIRSaveConfigCallback property) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup property) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup property) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup property) (lightning_ir.schedulers.schedulers.GenericScheduler property) step() (lightning_ir.schedulers.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.LinearLRSchedulerWithLinearWarmup method) (lightning_ir.schedulers.lr_schedulers.WarmupLRScheduler method) strict_loading (lightning_ir.base.module.LightningIRModule property) (lightning_ir.bi_encoder.module.BiEncoderModule property) (lightning_ir.cross_encoder.module.CrossEncoderModule property) SupervisedMarginMSE (class in lightning_ir.loss.loss)

T

T5CrossEncoderConfig (class in lightning_ir.models.t5.config) T5CrossEncoderModel (class in lightning_ir.models.t5.model) T5CrossEncoderTokenizer (class in lightning_ir.models.t5.tokenizer) teardown() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.callbacks.callbacks.IndexCallback method) (lightning_ir.callbacks.callbacks.RankCallback method) (lightning_ir.callbacks.callbacks.RegisterLocalDatasetCallback method) (lightning_ir.callbacks.callbacks.ReRankCallback method) (lightning_ir.callbacks.callbacks.SearchCallback method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) (lightning_ir.main.LightningIRSaveConfigCallback method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericConstantSchedulerWithQuadraticWarmup method) (lightning_ir.schedulers.schedulers.GenericLinearSchedulerWithLinearWarmup method) (lightning_ir.schedulers.schedulers.GenericScheduler method) test() (lightning_ir.main.LightningIRTrainer method) test_dataloader() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) test_dataloaders (lightning_ir.main.LightningIRTrainer property) test_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) to() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.BiEncoderEmbedding method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) to_added_args_dict() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) to_dict() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) to_diff_dict() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) to_empty() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) to_json_file() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method)

to_json_string() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) to_onnx() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) to_tokenizer_dict() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) to_torchscript() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) to_tuple() (lightning_ir.base.model.LightningIROutput method) (lightning_ir.bi_encoder.model.BiEncoderOutput method) (lightning_ir.cross_encoder.model.CrossEncoderOutput method) toggle_optimizer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) tokenize() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) tokenize_doc() (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) tokenize_query() (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) TOKENIZER_ARGS (lightning_ir.base.config.LightningIRConfig attribute) (lightning_ir.bi_encoder.config.BiEncoderConfig attribute) top() (lightning_ir.data.dataset.Sampler static method) top_and_random() (lightning_ir.data.dataset.Sampler static method) train() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method) train_dataloader (lightning_ir.main.LightningIRTrainer property) train_dataloader() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) TrainBatch (class in lightning_ir.data.data) training_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) transfer_batch_to_device() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) truncate_sequences() (lightning_ir.base.tokenizer.LightningIRTokenizer method) (lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer method) (lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer method) (lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer method) TupleDataset (class in lightning_ir.data.dataset) type() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

U

unfreeze() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) untoggle_optimizer() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) update() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method) update_from_string() (lightning_ir.base.config.LightningIRConfig method) (lightning_ir.bi_encoder.config.BiEncoderConfig method) (lightning_ir.cross_encoder.config.CrossEncoderConfig method) (lightning_ir.models.col.ColConfig method) (lightning_ir.models.splade.SpladeConfig method) (lightning_ir.models.t5.config.T5CrossEncoderConfig method) (lightning_ir.models.xtr.config.XTRConfig method)

use_artifact() (lightning_ir.main.LightningIRWandbLogger method) use_return_dict (lightning_ir.base.config.LightningIRConfig property) (lightning_ir.bi_encoder.config.BiEncoderConfig property) (lightning_ir.cross_encoder.config.CrossEncoderConfig property) (lightning_ir.models.col.ColConfig property) (lightning_ir.models.splade.SpladeConfig property) (lightning_ir.models.t5.config.T5CrossEncoderConfig property) (lightning_ir.models.xtr.config.XTRConfig property)

V

val_dataloader() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.data.datamodule.LightningIRDataModule method) val_dataloaders (lightning_ir.main.LightningIRTrainer property) validate() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.main.LightningIRTrainer method)

validate_loss() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) validate_metrics() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) validation_step() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) version (lightning_ir.main.LightningIRWandbLogger property)

W

WarmupLRScheduler (class in lightning_ir.schedulers.lr_schedulers)

X

xpu() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)

XTRConfig (class in lightning_ir.models.xtr.config) XTRModel (class in lightning_ir.models.xtr.model) XTRScoringFunction (class in lightning_ir.models.xtr.model)

Z

zero_grad() (lightning_ir.base.module.LightningIRModule method) (lightning_ir.bi_encoder.model.MLMHead method) (lightning_ir.bi_encoder.model.ScoringFunction method) (lightning_ir.bi_encoder.module.BiEncoderModule method) (lightning_ir.cross_encoder.module.CrossEncoderModule method) (lightning_ir.models.t5.model.ScaleLinear method) (lightning_ir.models.xtr.model.XTRScoringFunction method)