cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessorConfig#

class cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessorConfig(*args, **kwargs)[source]#

Bases: cerebras.modelzoo.data.vision.segmentation.UNetDataProcessor.UNetDataProcessorConfig

Methods

`check_for_deprecated_fields`
`check_literal_discriminator_field`
`copy`
`get_orig_class`
`get_orig_class_args`
`model_copy`
`model_post_init`
`post_init`
`set_convert_to_onehot`

Attributes

`augment_data`
`batch_size`
`class_id`
`convert_to_onehot`
`data_dir`
`discriminator`
`discriminator_value`
`drop_last`
`duplicate_act_worker_data`
`fp16_type`
`image_shape`
`loss`
`max_image_shape`
`mixed_precision`
`model_config`
`normalize_data_method`
`num_classes`
`num_workers`
`persistent_workers`
`prefetch_factor`
`shuffle`
`shuffle_seed`
`split`	Dataset split.
`train_test_split`
`use_fast_dataloader`
`use_worker_cache`
`data_processor`

split = 'train'#: Dataset split.

cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessor

cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset