datamodule

DataModule for Lightning IR that handles batching and collation of data samples.

This module defines the LightningIRDataModule class that handles batching and collation of data samples for training and inference in Lightning IR.

Classes

LightningIRDataModule([train_dataset, ...])