composer.datasets.dataset_registry#
Mapping between dataset names and corresponding HParams classes.
Functions
Returns a mapping between different supported datasets and their HParams classes that create an instance of the dataset. |
- composer.datasets.dataset_registry.get_dataset_registry()[source]#
Returns a mapping between different supported datasets and their HParams classes that create an instance of the dataset. An example entry in the returned dictionary:
"imagenet": ImagenetDatasetHparams
.- Returns
Dict[str, DatasetHparams] โ A dictionary of mapping.