cerebras.modelzoo.data.vision.classification.dataset_factory.VisionSubset

class cerebras.modelzoo.data.vision.classification.dataset_factory.VisionSubset

Bases: torch.utils.data.Subset

Methods

set_transforms	transforms (callable, optional): A function/transforms that takes in
truncate_to_idx

__init__(dataset, indices)

set_transforms(transforms=None, transform=None, target_transform=None)

transforms (callable, optional): A function/transforms that takes in

an image and a label and returns the transformed versions of both.

transform (callable, optional): A function/transform that takes in an PIL image

and returns a transformed version. E.g, transforms.RandomCrop

target_transform (callable, optional): A function/transform that takes in the

target and transforms it.

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

static __new__(cls, *args: Any, **kwargs: Any) → Any

previous

cerebras.modelzoo.data.vision.classification.dataset_factory.Processor

next

cerebras.modelzoo.data.vision.classification.mixup