TorchMnistDataset#

class TorchMnistDataset[source]#

MNIST Dataset.

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

static __new__(cls, *args, **kwds)#

Attributes

`TorchMnistDataset.class_to_idx`
`TorchMnistDataset.classes`
`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`()

previous

MnistModel.zero_grad

next

TorchMnistDataset.class_to_idx