if statement to elif statement

Christinaliang · Feb 13, 2019 · c8c5bb6 · c8c5bb6
1 parent f53f080
commit c8c5bb6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/5-2.BERT/BERT-Torch.py b/5-2.BERT/BERT-Torch.py
@@ -171,7 +171,7 @@ def make_batch():
             masked_tokens.append(input_ids[pos])
             if random() < 0.8:  # 80%
                 input_ids[pos] = word_dict['[MASK]'] # make mask
-            if random() < 0.5:  # 10%
+            elif random() < 0.5:  # 10%
                 index = randint(0, vocab_size - 1) # random index in vocabulary
                 input_ids[pos] = word_dict[number_dict[index]] # replace
 

diff --git a/5-2.BERT/BERT_Torch.ipynb b/5-2.BERT/BERT_Torch.ipynb
@@ -200,7 +200,7 @@
         "            masked_tokens.append(input_ids[pos])\n",
         "            if random() < 0.8:  # 80%\n",
         "                input_ids[pos] = word_dict['[MASK]'] # make mask\n",
-        "            if random() < 0.5:  # 10%\n",
+        "            elif random() < 0.5:  # 10%\n",
         "                index = randint(0, vocab_size - 1) # random index in vocabulary\n",
         "                input_ids[pos] = word_dict[number_dict[index]] # replace\n",
         "\n",