cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor.ImageNet21KMAEProcessor#
- class cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor.ImageNet21KMAEProcessor(params)[source]#
Bases:
cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor.MAEProcessor
,cerebras.modelzoo.data.vision.classification.data.imagenet21k.ImageNet21KProcessor
Methods
check_split_valid
create_dataloader
create_dataset
create_shuffled_idx
generate_masks
generate_partition
mae_collate_fn
process_transform
split_dataset