cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor.ImageNet21KMAEProcessor#

class cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor.ImageNet21KMAEProcessor(config)[source]#

Bases: cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor.MAEProcessor, cerebras.modelzoo.data.vision.classification.data.imagenet21k.ImageNet21KProcessor

Methods

create_dataloader

create_dataset

create_shuffled_idx

generate_masks

generate_partition

mae_collate_fn

process_transform

split_dataset

previous

cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor

next

cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor.ImageNet21KMAEProcessorConfig