cerebras.modelzoo.data.vision.classification.data.svhn.SVHNProcessorConfig#
- class cerebras.modelzoo.data.vision.classification.data.svhn.SVHNProcessorConfig(*args, **kwargs)[source]#
Bases:
cerebras.modelzoo.data.vision.classification.dataset_factory.VisionClassificationProcessorConfig
Methods
check_for_deprecated_fields
check_literal_discriminator_field
copy
get_orig_class
get_orig_class_args
model_copy
model_post_init
post_init
Attributes
batch_size
Global batch size for the dataloader
cutmix_alpha
Alpha parameter for the cutmix transform.
data_dir
The path to the data
discriminator
discriminator_value
drop_last
Similar to the PyTorch drop_last setting except that samples that when set to True, samples that would have been dropped at the end of one epoch are yielded at the start of the next epoch so that there is no data loss.
fp16_type
image_size
The size of the images in the dataset
mixed_precision
mixup_alpha
Alpha parameter for the mixup transform.
model_config
noaugment
Indicates to skip augmentation as part of preprocessing.
num_classes
The number of classification classes in the dataset
num_workers
How many subprocesses to use for data loading
persistent_workers
Whether or not to keep workers persistent between epochs.
prefetch_factor
Number of batches loaded in advance by each worker
ra_sampler_num_repeat
Number of repeats for Repeated Augmentation sampler.
sampler
Type of data sampler to use
shuffle
Whether or not to shuffle the dataset.
shuffle_seed
The seed used for deterministic shuffling.
Dataset split.
transforms
List of transforms for preprocessing
use_worker_cache
data_processor
- split = 'train'#
Dataset split.