SyntheticDatasetHparams

class composer.datasets.SyntheticDatasetHparams(total_dataset_size: int, data_shape: List[int], num_unique_samples_to_create: int = 100, data_type: composer.datasets.synthetic.SyntheticDataType = SyntheticDataType.GAUSSIAN, label_type: composer.datasets.synthetic.SyntheticDataLabelType = SyntheticDataLabelType.CLASSIFICATION_INT, num_classes: Optional[int] = 2, label_shape: List[int] = <factory>, device: str = 'cpu', memory_format: composer.datasets.synthetic.MemoryFormat = MemoryFormat.CONTIGUOUS_FORMAT, drop_last: bool = True, shuffle: bool = True)

Bases: composer.datasets.hparams.DatasetHparams

Defines an instance of a synthetic dataset for classification.

See SyntheticDataset

initialize_object() → DataloaderSpec

Initializes a DataloaderSpec for this dataset.

validate()

Validate that the hparams are of the correct types. Recurses through sub-hparams.

Raises

TypeError – Raises a TypeError if any fields are an incorrect type.