cerebras.modelzoo.data.vision.masked_auto_encoding.config.MAEProcessorConfig#
- class cerebras.modelzoo.data.vision.masked_auto_encoding.config.MAEProcessorConfig(batch_size: int = <object object at 0x7fc286331b70>, shuffle: bool = True, shuffle_seed: int = 0, num_workers: int = 0, prefetch_factor: Optional[int] = None, persistent_workers: Optional[bool] = None, image_size: List[int] = <factory>, patch_size: List[int] = <factory>, image_channels: List[int] = <object object at 0x7fc286331b70>, mask_ratio: float = 0.75)[source]#
- image_size: List[int]#
- patch_size: List[int]#
- image_channels: List[int] = <object object>#
- mask_ratio: float = 0.75#
- batch_size: int = <object object>#
Batch size to be used
- num_workers: int = 0#
The number of PyTorch processes used in the dataloader
- persistent_workers: Optional[bool] = None#
Whether or not to keep workers persistent between epochs
- prefetch_factor: Optional[int] = None#
The number of batches to prefetch in the dataloader
- shuffle: bool = True#
Whether or not to shuffle the dataset
- shuffle_seed: int = 0#
Seed used for deterministic shuffling