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

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

Bases: object

Methods

generate_masks

generate_partition

mae_collate_fn