cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataProcessorConfig

class cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataProcessorConfig(*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
class_id
convert_to_onehot
data_dir
disable_sharding
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
overfit
overfit_indices
overfit_num_batches
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.

previous

cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataProcessor

next

cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataset