From e1b3cfb5047c76cf91d118485040039f3ee7df93 Mon Sep 17 00:00:00 2001 From: Liang Niu Date: Tue, 12 Feb 2019 15:54:23 +0400 Subject: [PATCH] fixed a minor bug in README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index ac88314c21..52e8c518de 100644 --- a/README.md +++ b/README.md @@ -165,7 +165,7 @@ text = "[CLS] Who was Jim Henson ? [SEP] Jim Henson was a puppeteer [SEP]" tokenized_text = tokenizer.tokenize(text) # Mask a token that we will try to predict back with `BertForMaskedLM` -masked_index = 6 +masked_index = 8 tokenized_text[masked_index] = '[MASK]' assert tokenized_text == ['[CLS]', 'who', 'was', 'jim', 'henson', '?', '[SEP]', 'jim', '[MASK]', 'was', 'a', 'puppet', '##eer', '[SEP]']