main

Main entry point for Lightning IR using the Lightning IR CLI.

The module also defines several helper classes for configuring and running experiments.

Functions

main()

Entry point for the Lightning IR command line interface.

Classes

LightningIRCLI(model_class, ...)

Lightning IR Command Line Interface that extends PyTorch LightningCLI_ for information retrieval tasks.

LightningIRSaveConfigCallback(parser, config)

Lightning IR configuration saving callback with intelligent save conditions.

LightningIRTrainer(*[, accelerator, ...])

Lightning IR Trainer that extends PyTorch Lightning Trainer with information retrieval specific methods.

LightningIRWandbLogger([name, save_dir, ...])

Lightning IR extension of the Weights & Biases Logger for enhanced experiment tracking.