cerebras.modelzoo.data.vision.segmentation.config.SeverstalBinaryClassDataProcessorConfig#
- class cerebras.modelzoo.data.vision.segmentation.config.SeverstalBinaryClassDataProcessorConfig(batch_size: int = <object object at 0x7f2933a50b80>, shuffle: bool = True, shuffle_seed: int = 0, num_workers: int = 0, prefetch_factor: int = 10, persistent_workers: bool = True, data_dir: Union[str, List[str]] = <object object at 0x7f2933a50b80>, num_classes: int = <object object at 0x7f2933a50b80>, loss: str = <object object at 0x7f2933a50b80>, normalize_data_method: str = <object object at 0x7f2933a50b80>, augment_data: bool = True, drop_last: bool = True, mixed_precision: Optional[bool] = None, use_fast_dataloader: bool = False, duplicate_act_worker_data: bool = False, use_worker_cache: bool = <object object at 0x7f2933a50b80>, train_test_split: float = <object object at 0x7f2933a50b80>, class_id: int = <object object at 0x7f2933a50b80>, image_shape: List[int] = <factory>, max_image_shape: List[int] = <factory>)[source]#
- use_worker_cache: bool = <object object>#
- train_test_split: float = <object object>#
- class_id: int = <object object>#
- image_shape: List[int]#
- max_image_shape: List[int]#
- augment_data: bool = True#
- batch_size: int = <object object>#
Batch size to be used
- data_dir: Union[str, List[str]] = <object object>#
- drop_last: bool = True#
- duplicate_act_worker_data: bool = False#
- loss: str = <object object>#
- mixed_precision: Optional[bool] = None#
- normalize_data_method: str = <object object>#
- 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
- shuffle: bool = True#
Whether or not to shuffle the dataset
- shuffle_seed: int = 0#
Seed used for deterministic shuffling
- use_fast_dataloader: bool = False#