cerebras.modelzoo.data.nlp.bert.BertSumCSVDataProcessor#
Processor for PyTorch BERT fine tuning - Summarization.
Functions
Creates the feature dict for bertsum model after applying padding. |
Classes
Reads csv file containing the input_token_ids, and label_ids. |