cerebras.modelzoo.data.vision.segmentation.CityscapesDataProcessor.CityscapesDataProcessorConfig

class cerebras.modelzoo.data.vision.segmentation.CityscapesDataProcessor.CityscapesDataProcessorConfig(*args, **kwargs)

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
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.
use_fast_dataloader
use_worker_cache
data_processor

split = 'train'

Dataset split.

previous

cerebras.modelzoo.data.vision.segmentation.CityscapesDataProcessor.CityscapesDataProcessor

next

cerebras.modelzoo.data.vision.segmentation.Hdf5BaseDataProcessor