cerebras.modelzoo.data.vision.classification.data.imagenet21k.ImageNet21KProcessor#

class cerebras.modelzoo.data.vision.classification.data.imagenet21k.ImageNet21KProcessor(config)[source]#

Bases: cerebras.modelzoo.data.vision.classification.dataset_factory.VisionClassificationProcessor

Methods

create_dataloader

create_dataset

create_shuffled_idx

process_transform

split_dataset

previous

cerebras.modelzoo.data.vision.classification.data.imagenet21k.ImageNet21K

next

cerebras.modelzoo.data.vision.classification.data.imagenet21k.ImageNet21KProcessorConfig