cerebras.modelzoo.data.common#

GenericDataProcessor

Pytorch Generic Dataloader

HDF5IterableDataProcessor

Pytorch HDF5 Dataloader

HDF5IterableDataset

PyTorch HDF5 Dataset

SyntheticDataProcessor

Utilities for generating synthetic data based on some specification.

h5_map_dataset

input_utils

tensor_spec

Wrapper class used to process TensorSpecs using a custom yaml tag.

previous

cerebras.modelzoo.data

next

cerebras.modelzoo.data.common.GenericDataProcessor