cerebras.modelzoo.data.vision.masked_auto_encoding.config.ImageNet21KProcessorConfig#
- class cerebras.modelzoo.data.vision.masked_auto_encoding.config.ImageNet21KProcessorConfig(batch_size: int = <object object at 0x7f0436677b60>, shuffle: bool = True, shuffle_seed: int = 0, num_workers: int = 0, prefetch_factor: int = 10, persistent_workers: bool = True, image_size: int = 224, patch_size: List[int] = <factory>, image_channels: List[int] = <object object at 0x7f0436677b60>, mask_ratio: float = 0.75, data_dir: Union[str, List[str]] = <object object at 0x7f0436677b60>, num_classes: int = <object object at 0x7f0436677b60>, mixed_precision: bool = <object object at 0x7f0436677b60>, transforms: List[dict] = <factory>, noaugment: bool = False, drop_last: bool = True, sampler: str = 'random', ra_sampler_num_repeat: int = 3, mixup_alpha: float = 0.1, cutmix_alpha: float = 0.1)[source]#
- batch_size: int = <object object>#
Batch size to be used
- cutmix_alpha: float = 0.1#
- data_dir: Union[str, List[str]] = <object object>#
- drop_last: bool = True#
- image_channels: List[int] = <object object>#
- image_size: int = 224#
- mask_ratio: float = 0.75#
- mixed_precision: bool = <object object>#
- mixup_alpha: float = 0.1#
- noaugment: bool = False#
- num_classes: int = <object object>#
- num_workers: int = 0#
The number of PyTorch processes used in the dataloader
- persistent_workers: bool = True#
Whether or not to keep workers persistent between epochs
- prefetch_factor: int = 10#
The number of batches to prefetch in the dataloader
- ra_sampler_num_repeat: int = 3#
- sampler: str = 'random'#
- shuffle: bool = True#
Whether or not to shuffle the dataset
- shuffle_seed: int = 0#
Seed used for deterministic shuffling
- transforms: List[dict]#
- patch_size: List[int]#