Thank you for your feedback!
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
Was this information helpful?
Thank you for your feedback!
- NO
- YES
Cancel
Submit