cerebras.modelzoo.data_preparation.nlp.bert.sentence_pair_processor

Functions

data_generator

Generator function used to create input dataset for MLM + NSP dataset.

Classes

SentencePairInstance

A single training (sentence-pair) instance.

previous

cerebras.modelzoo.data_preparation.nlp.bert.parser_utils.get_parser_args

next

cerebras.modelzoo.data_preparation.nlp.bert.sentence_pair_processor.data_generator