if statement to elif statement

Author: kimjiwon

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
 

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",