cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.SST2DataProcessor#

class cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.SST2DataProcessor[source]#

Bases: cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.DataProcessor

The data processor responsible for creating the SST2 dataloader instance.

Methods

create_dataloader

create_dataset

__init__(data_params, model_params) → None[source]#

previous

cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.MNLIDataset

next

cerebras.modelzoo.data.nlp.bert.BertClassifierDataProcessor.SST2Dataset