cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor.MAEProcessor#

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

Bases: object

Methods

generate_masks

generate_partition

mae_collate_fn

previous

cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor

next

cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor.MAEProcessorConfig