cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.MNLIDataProcessorConfig#
- class cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.MNLIDataProcessorConfig(*args, **kwargs)[source]#
Bases:
cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.ClassifierDataProcessorConfigMethods
check_for_deprecated_fieldscheck_literal_discriminator_fieldcopyget_orig_classget_orig_class_argsget_vocab_filemodel_copymodel_post_initpost_initAttributes
attn_mask_pad_idbatch_sizeThe batch size.
data_dirPath to the data files to use.
discriminatordiscriminator_valuedo_lowerFlag to lower case the texts.
drop_lastWhether to drop last batch of epoch if it's an incomplete batch.
input_pad_idis_trainingWhether the data processor is used for training or validation.
labels_pad_idmax_sequence_lengthmodel_confignum_workersThe number of PyTorch processes used in the dataloader.
persistent_workersWhether or not to keep workers persistent between epochs.
prefetch_factorThe number of batches to prefetch in the dataloader.
shuffleWhether or not to shuffle the dataset.
shuffle_seedThe seed used for deterministic shuffling.
vocab_filePath to the vocabulary file.
data_processor