CIFAR10DatasetHparams

class composer.datasets.CIFAR10DatasetHparams(is_train: bool, datadir: str, download: bool, drop_last: bool = True, shuffle: bool = True)[source]

Bases: composer.datasets.hparams.DatasetHparams

Defines an instance of the CIFAR-10 dataset for image classification.

Parameters

  • is_train (bool) – Whether to load the training or validation dataset.

  • datadir (str) – Data directory to use.

  • download (bool) – Whether to download the dataset, if needed.

  • drop_last (bool) – Whether to drop the last samples for the last batch.

  • shuffle (bool) – Whether to shuffle the dataset for each epoch.

initialize_object() → DataloaderSpec[source]

Initializes a DataloaderSpec for this dataset.