model

Model module for Lightning IR.

This module contains the main model class and output class for the Lightning IR library.

Functions

batch_encoding_wrapper(func)

Decorator to enable sub-batching for models that support it.

Classes

BatchEncodingWrapper(*args, **kwargs)
LightningIRModel(config, *args, **kwargs)	Base class for Lightning IR models.
LightningIROutput([scores])	Base class for the output of the Lightning IR model.