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

Lightning IR module for loss functions.

main

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

modeling_utils

models

Lightning IR module for native models.

retrieve

schedulers

Schedulers for adjusting lr and generic values during fine-tuning.