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

cerebras.modelzoo.data_preparation.nlp.bert.mlm_only_processor.create_masked_lm_features(example, vocab_words, max_seq_length, mask_whole_word, max_predictions_per_seq, masked_lm_prob, document_separator_token, rng, tokenizer, output_type_shapes, inverted_mask)[source]#

previous

cerebras.modelzoo.data_preparation.nlp.bert.mlm_only_processor

next

cerebras.modelzoo.data_preparation.nlp.bert.mlm_only_processor.data_generator