composer.datasets.dataset_registry

composer.datasets.dataset_registry

Functions

get_dataset_registry

composer.datasets.dataset_registry.get_dataset_registry

Hparams

These classes are used with yahp for YAML-based configuration.

ADE20kDatasetHparams	Defines an instance of the ADE20k dataset for semantic segmentation.
BratsDatasetHparams	Defines an instance of the BraTS dataset for image segmentation.
CIFAR10DatasetHparams	Defines an instance of the CIFAR-10 dataset for image classification.
GLUEHparams	Sets up a generic GLUE dataset loader.
ImagenetDatasetHparams	Defines an instance of the ImageNet dataset for image classification.
LMDatasetHparams	Defines a generic dataset class for autoregressive and masked language models trained with self-supervised learning.
MNISTDatasetHparams	Defines an instance of the MNIST dataset for image classification.

Attributes

• registry