nip.image_classification.data.Cifar10DatasetWrapper

nip.image_classification.data.Cifar10DatasetWrapper#

class nip.image_classification.data.Cifar10DatasetWrapper(root: str | Path, train: bool = True, transform: Callable | None = None, target_transform: Callable | None = None, download: bool = False)[source]#

The CIFAR-10 dataset wrapper.

Attributes

base_folder

binarification_seed

filename

height

meta

num_channels

selected_classes

test_list

tgz_md5

train_list

url

width

Methods