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

class cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor.ImageNet21KMAEProcessor[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`

__init__(params)[source]#

previous

cerebras.modelzoo.data.vision.masked_auto_encoding.ImageNet21KMAEProcessor

next

cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor