Skip to content
Lightning IR
Toggle navigation menu
⌘
K
Lightning IR
/
Index
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.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.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.DataParallelIterableDataset method)
(lightning_ir.data.dataset.DocDataset method)
(lightning_ir.data.dataset.IRDataset 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.lightning_utils.callbacks.GatherMixin method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.LinearLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.WarmupLRScheduler method)
(lightning_ir.lightning_utils.schedulers.ConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.ConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.lightning_utils.schedulers.LambdaWarmupScheduler method)
(lightning_ir.lightning_utils.schedulers.LinearSchedulerWithLinearWarmup 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.KLDivergence method)
(lightning_ir.loss.loss.L1Regularization method)
(lightning_ir.loss.loss.L2Regularization method)
(lightning_ir.loss.loss.ListwiseLossFunction method)
(lightning_ir.loss.loss.LocalizedContrastiveEstimation 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.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.faiss_indexer.FaissFlatIndexConfig method)
(lightning_ir.retrieve.faiss_indexer.FaissFlatIndexer method)
(lightning_ir.retrieve.faiss_indexer.FaissIndexConfig method)
(lightning_ir.retrieve.faiss_indexer.FaissIndexer method)
(lightning_ir.retrieve.faiss_indexer.FaissIVFIndexConfig method)
(lightning_ir.retrieve.faiss_indexer.FaissIVFIndexer method)
(lightning_ir.retrieve.faiss_indexer.FaissIVFPQIndexConfig method)
(lightning_ir.retrieve.faiss_indexer.FaissIVFPQIndexer method)
(lightning_ir.retrieve.faiss_searcher.FaissSearchConfig method)
(lightning_ir.retrieve.faiss_searcher.FaissSearcher method)
(lightning_ir.retrieve.indexer.IndexConfig method)
(lightning_ir.retrieve.indexer.Indexer method)
(lightning_ir.retrieve.searcher.SearchConfig method)
(lightning_ir.retrieve.searcher.Searcher method)
(lightning_ir.retrieve.sparse_indexer.SparseIndexConfig method)
(lightning_ir.retrieve.sparse_indexer.SparseIndexer method)
(lightning_ir.retrieve.sparse_searcher.SparseIndex method)
(lightning_ir.retrieve.sparse_searcher.SparseSearchConfig method)
(lightning_ir.retrieve.sparse_searcher.SparseSearcher 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)
ADDED_ARGS (lightning_ir.base.config.LightningIRConfig attribute)
all_gather() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
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)
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)
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)
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)
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)
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)
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)
compute_losses() (lightning_ir.base.module.LightningIRModule 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)
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.lightning_utils.lr_schedulers)
ConstantMarginMSE (class in lightning_ir.loss.loss)
ConstantSchedulerWithLinearWarmup (class in lightning_ir.lightning_utils.schedulers)
ConstantSchedulerWithQuadraticWarmup (class in lightning_ir.lightning_utils.schedulers)
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)
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
DataParallelIterableDataset (class in lightning_ir.data.dataset)
default_root_dir (lightning_ir.main.LightningIRTrainer 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)
doc_ids (lightning_ir.data.ir_datasets_utils.ScoredDocTuple attribute)
DocDataset (class in lightning_ir.data.dataset)
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)
enable_validation (lightning_ir.main.LightningIRTrainer property)
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_indexer)
FaissFlatIndexer (class in lightning_ir.retrieve.faiss_indexer)
FaissIndexConfig (class in lightning_ir.retrieve.faiss_indexer)
FaissIndexer (class in lightning_ir.retrieve.faiss_indexer)
FaissIVFIndexConfig (class in lightning_ir.retrieve.faiss_indexer)
FaissIVFIndexer (class in lightning_ir.retrieve.faiss_indexer)
FaissIVFPQIndexConfig (class in lightning_ir.retrieve.faiss_indexer)
FaissIVFPQIndexer (class in lightning_ir.retrieve.faiss_indexer)
FaissSearchConfig (class in lightning_ir.retrieve.faiss_searcher)
FaissSearcher (class in lightning_ir.retrieve.faiss_searcher)
fit() (lightning_ir.main.LightningIRTrainer method)
flash_attention_forward() (in module lightning_ir.flash.flash_bert)
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)
format_large_number() (in module lightning_ir.lightning_utils.callbacks)
forward() (lightning_ir.base.model.LightningIRModel method)
(lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.model.ScoringFunction 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_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.model.LightningIRModel class method)
(lightning_ir.base.tokenizer.LightningIRTokenizer class method)
(lightning_ir.bi_encoder.model.BiEncoderModel class method)
(lightning_ir.bi_encoder.tokenizer.BiEncoderTokenizer class method)
(lightning_ir.cross_encoder.model.CrossEncoderModel class method)
(lightning_ir.cross_encoder.tokenizer.CrossEncoderTokenizer class method)
(lightning_ir.models.col.ColModel class method)
(lightning_ir.models.splade.SpladeModel class method)
(lightning_ir.models.t5.model.T5CrossEncoderModel class method)
(lightning_ir.models.t5.tokenizer.T5CrossEncoderTokenizer class method)
fromkeys() (lightning_ir.base.model.LightningIROutput method)
(lightning_ir.bi_encoder.model.BiEncoderOutput method)
(lightning_ir.cross_encoder.model.CrossEncoderOutput method)
G
GatherMixin (class in lightning_ir.lightning_utils.callbacks)
GenericConstantSchedulerWithLinearWarmup (class in lightning_ir.lightning_utils.schedulers)
GenericConstantSchedulerWithQuadraticWarmup (class in lightning_ir.lightning_utils.schedulers)
GenericLinearSchedulerWithLinearWarmup (class in lightning_ir.lightning_utils.schedulers)
GenericScheduler (class in lightning_ir.lightning_utils.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_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.lightning_utils.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.LinearLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.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_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_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)
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.lightning_utils.callbacks)
IndexConfig (class in lightning_ir.retrieve.indexer)
Indexer (class in lightning_ir.retrieve.indexer)
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)
IRDataset (class in lightning_ir.data.dataset)
is_global_zero (lightning_ir.main.LightningIRTrainer property)
is_last_batch (lightning_ir.main.LightningIRTrainer property)
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.lightning_utils.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.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.flash
module
lightning_ir.flash.flash_bert
module
lightning_ir.lightning_utils
module
lightning_ir.lightning_utils.callbacks
module
lightning_ir.lightning_utils.lr_schedulers
module
lightning_ir.lightning_utils.schedulers
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.faiss_indexer
module
lightning_ir.retrieve.faiss_searcher
module
lightning_ir.retrieve.indexer
module
lightning_ir.retrieve.searcher
module
lightning_ir.retrieve.sparse_indexer
module
lightning_ir.retrieve.sparse_searcher
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.lightning_utils.lr_schedulers)
LinearSchedulerWithLinearWarmup (class in lightning_ir.lightning_utils.schedulers)
link_optimizers_and_lr_schedulers() (lightning_ir.main.LightningIRCLI static method)
ListwiseLossFunction (class in lightning_ir.loss.loss)
load_constituent() (in module lightning_ir.data.ir_datasets_utils)
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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.data.datamodule.LightningIRDataModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.LinearLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.WarmupLRScheduler method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
(lightning_ir.models.t5.model.ScaleLinear method)
(lightning_ir.models.xtr.model.XTRScoringFunction method)
local_rank (lightning_ir.base.module.LightningIRModule property)
(lightning_ir.bi_encoder.module.BiEncoderModule property)
(lightning_ir.cross_encoder.module.CrossEncoderModule property)
LocalizedContrastiveEstimation (class in lightning_ir.loss.loss)
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_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)
MLMHead (class in lightning_ir.bi_encoder.model)
model (lightning_ir.main.LightningIRTrainer property)
model_type (lightning_ir.base.config.LightningIRConfig 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.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.flash
lightning_ir.flash.flash_bert
lightning_ir.lightning_utils
lightning_ir.lightning_utils.callbacks
lightning_ir.lightning_utils.lr_schedulers
lightning_ir.lightning_utils.schedulers
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.faiss_indexer
lightning_ir.retrieve.faiss_searcher
lightning_ir.retrieve.indexer
lightning_ir.retrieve.searcher
lightning_ir.retrieve.sparse_indexer
lightning_ir.retrieve.sparse_searcher
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)
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_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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_before_zero_grad() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_exception() (lightning_ir.data.datamodule.LightningIRDataModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_fit_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_fit_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_predict_batch_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_predict_batch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_predict_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_predict_epoch_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_predict_epoch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_sanity_check_end() (lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_sanity_check_start() (lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_save_checkpoint() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_test_batch_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_test_batch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_test_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_test_epoch_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_test_epoch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_train_batch_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_train_batch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_train_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_train_epoch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_train_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_validation_batch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_validation_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_validation_epoch_end() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
on_validation_epoch_start() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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
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)
popitem() (lightning_ir.base.model.LightningIROutput method)
(lightning_ir.bi_encoder.model.BiEncoderOutput method)
(lightning_ir.cross_encoder.model.CrossEncoderOutput method)
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_input() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule 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.lightning_utils.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.LinearLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.WarmupLRScheduler method)
progress_bar_callback (lightning_ir.main.LightningIRTrainer property)
progress_bar_metrics (lightning_ir.main.LightningIRTrainer property)
Q
query_id (lightning_ir.data.ir_datasets_utils.ScoredDocTuple attribute)
QueryDataset (class in lightning_ir.data.dataset)
QuerySample (class in lightning_ir.data.data)
R
RankBatch (class in lightning_ir.data.data)
RankCallback (class in lightning_ir.lightning_utils.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_colbert_docpairs() (in module lightning_ir.data.ir_datasets_utils)
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_kd_docpairs() (in module lightning_ir.data.ir_datasets_utils)
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_local() (in module lightning_ir.data.ir_datasets_utils)
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_rank_distillm() (in module lightning_ir.data.ir_datasets_utils)
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)
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.lightning_utils.callbacks)
root_dir (lightning_ir.main.LightningIRWandbLogger property)
RunDataset (class in lightning_ir.data.dataset)
S
Sampler (class in lightning_ir.data.dataset)
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.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
ScaleLinear (class in lightning_ir.models.t5.model)
score() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
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)
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.lightning_utils.callbacks)
SearchConfig (class in lightning_ir.retrieve.searcher)
Searcher (class in lightning_ir.retrieve.searcher)
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)
setup() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.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)
SparseIndex (class in lightning_ir.retrieve.sparse_searcher)
SparseIndexConfig (class in lightning_ir.retrieve.sparse_indexer)
SparseIndexer (class in lightning_ir.retrieve.sparse_indexer)
SparseSearchConfig (class in lightning_ir.retrieve.sparse_searcher)
SparseSearcher (class in lightning_ir.retrieve.sparse_searcher)
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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.data.datamodule.LightningIRDataModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.lr_schedulers.ConstantLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.LinearLRSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.lr_schedulers.WarmupLRScheduler method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback method)
(lightning_ir.models.t5.model.ScaleLinear method)
(lightning_ir.models.xtr.model.XTRScoringFunction method)
state_key (lightning_ir.lightning_utils.callbacks.IndexCallback property)
(lightning_ir.lightning_utils.callbacks.RankCallback property)
(lightning_ir.lightning_utils.callbacks.ReRankCallback property)
(lightning_ir.lightning_utils.callbacks.SearchCallback property)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup property)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup property)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup property)
(lightning_ir.lightning_utils.schedulers.GenericScheduler property)
(lightning_ir.main.LightningIRSaveConfigCallback property)
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.cross_encoder.module.CrossEncoderModule method)
(lightning_ir.data.datamodule.LightningIRDataModule method)
(lightning_ir.lightning_utils.callbacks.IndexCallback method)
(lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericConstantSchedulerWithQuadraticWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericLinearSchedulerWithLinearWarmup method)
(lightning_ir.lightning_utils.schedulers.GenericScheduler method)
(lightning_ir.main.LightningIRSaveConfigCallback 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)
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.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.cross_encoder.config.CrossEncoderConfig method)
(lightning_ir.models.t5.config.T5CrossEncoderConfig 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_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)
TOKENIZER_ARGS (lightning_ir.base.config.LightningIRConfig attribute)
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)
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)
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)
use_artifact() (lightning_ir.main.LightningIRWandbLogger method)
V
val_dataloader() (lightning_ir.base.module.LightningIRModule method)
(lightning_ir.bi_encoder.module.BiEncoderModule method)
(lightning_ir.cross_encoder.module.CrossEncoderModule 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.cross_encoder.module.CrossEncoderModule method)
version (lightning_ir.main.LightningIRWandbLogger property)
W
WarmupLRScheduler (class in lightning_ir.lightning_utils.lr_schedulers)
write_on_epoch_end() (lightning_ir.lightning_utils.callbacks.RankCallback method)
(lightning_ir.lightning_utils.callbacks.ReRankCallback method)
(lightning_ir.lightning_utils.callbacks.SearchCallback method)
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)