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

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

Bases: object

Methods

generate_masks

generate_partition

mae_collate_fn

__init__(params)[source]#

previous

cerebras.modelzoo.data.vision.masked_auto_encoding.MAEProcessor

next

cerebras.modelzoo.data.vision.preprocessing