torchpack.datasets.vision package

Submodules

torchpack.datasets.vision.cifar module

class torchpack.datasets.vision.cifar.CIFAR(*, root: str, num_classes: int = 10, transforms: Optional[Dict[str, Callable]] = None)

Bases: torchpack.datasets.dataset.Dataset

torchpack.datasets.vision.imagenet module

class torchpack.datasets.vision.imagenet.ImageNet(*, root: str, num_classes: int = 1000, transforms: Optional[Dict[str, Callable]] = None)

Bases: torchpack.datasets.dataset.Dataset

Module contents