cerebras.modelzoo.data.nlp.bert.BertCSVDataProcessor

Processor for PyTorch BERT training.

Classes

BertCSVDataProcessor

Reads csv files containing the input text tokens, and MLM features.

previous

cerebras.modelzoo.data.nlp.bert

next

cerebras.modelzoo.data.nlp.bert.BertCSVDataProcessor.BertCSVDataProcessor