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

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

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

The data processor responsible for creating the SST2 dataloader instance.

Methods

create_dataloader

create_dataset

previous

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

next

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