cerebras.modelzoo.data_preparation.nlp.bert.mlm_only_processor

Functions

create_masked_lm_features
data_generator	Generator function used to create input dataset for MLM only dataset.

Classes

MLMOnlyInstance

A single training MLMOnly instance.

previous

cerebras.modelzoo.data_preparation.nlp.bert.fine_tuning.token_classifier

next

cerebras.modelzoo.data_preparation.nlp.bert.mlm_only_processor.create_masked_lm_features