mixmo.loaders.cifar_dataset.CustomCIFAR10

class mixmo.loaders.cifar_dataset.CustomCIFAR10(root, train=True, transform=None, target_transform=None, download=False)

Bases: torchvision.datasets.cifar.CIFAR10

Torchvision’s CIFAR10 dataset class augmented with a custom __getitem__ method

__init__(root, train=True, transform=None, target_transform=None, download=False)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(root[, train, transform, …])	Initialize self.
download()
extra_repr()

Attributes

base_folder
filename
meta
test_list
tgz_md5
train_list
url