xlstm_jax.trainer.data_module

Contents

xlstm_jax.trainer.data_module#

Attributes#

Classes#

DataloaderModule

Module Contents#

xlstm_jax.trainer.data_module.DataIterator#

class xlstm_jax.trainer.data_module.DataloaderModule#

train_dataloader: DataIterator | None = None#

val_dataloader: DataIterator | dict[str, DataIterator] | None = None#

test_dataloader: DataIterator | dict[str, DataIterator] | None = None#