TorchMnistDataset#

class TorchMnistDataset[source]#

MNIST Dataset.

__init__(root: str, train: bool = True, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False) → None#

__new__(*args, **kwargs)#

Attributes

`TorchMnistDataset.class_to_idx`
`TorchMnistDataset.classes`
`TorchMnistDataset.functions`
`TorchMnistDataset.mirrors`
`TorchMnistDataset.processed_folder`
`TorchMnistDataset.raw_folder`	Return the path to the raw data folder.
`TorchMnistDataset.resources`
`TorchMnistDataset.test_data`
`TorchMnistDataset.test_file`
`TorchMnistDataset.test_labels`
`TorchMnistDataset.train_data`
`TorchMnistDataset.train_labels`
`TorchMnistDataset.training_file`

Methods

`TorchMnistDataset.download`()	Download the MNIST data if it doesn't exist already.
`TorchMnistDataset.extra_repr`()
`TorchMnistDataset.register_datapipe_as_function`(...)
`TorchMnistDataset.register_function`(...)

previous

MnistModel.zero_grad

next

TorchMnistDataset.class_to_idx