lightning_ir

Modules

base

Lightning IR base module.

bi_encoder

Base module for bi-encoder models.

callbacks

Module containing Lightning IR callbacks.

cross_encoder

Base module for cross-encoder models.

data

Lightning IR data module.

loss

main

models

retrieve

schedulers

Module containing utility classes and functions for PyTorch Lightning.