nip.scenario_base.data

nip.scenario_base.data#

Base classes for handling data.

Dataloaders yield tensordicts.

Classes

Dataset(hyper_params, settings, protocol_handler)

Base class for all datasets.

NestedArrayDictDataLoader(dataset[, ...])

Dataloader class for NestedArrayDict datasets.

NestedArrayDictDataset(hyper_params, ...[, ...])

Base class for all datasets based on nested array dicts.

TensorDictDataLoader(dataset, **kwargs)

Dataloader class for TensorDict datasets.

TensorDictDataset(hyper_params, settings, ...)

Base class for datasets based on tensordicts.

TensorDictSingleProcessDataLoaderIter(loader)

Single process DataLoaderIter for tensordicts.

Exceptions

CachedPretrainedEmbeddingsNotFound(model_name)

Raised when the cached embeddings for a pretrained model are not found.