cerebras.modelzoo.data.vision.segmentation.UNetDataProcessor.UNetDataProcessorConfig#

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

Bases: cerebras.modelzoo.config.data_config.DataConfig

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

convert_to_onehot

data_dir

discriminator

discriminator_value

drop_last

duplicate_act_worker_data

fp16_type

image_shape

loss

mixed_precision

model_config

normalize_data_method

num_classes

num_workers

persistent_workers

prefetch_factor

shuffle

shuffle_seed

use_fast_dataloader

data_processor