From e34e8febff1858abe897dced33c9d5c975b66bc1 Mon Sep 17 00:00:00 2001 From: Ningchen Wang Date: Sat, 14 Aug 2021 01:34:22 +0800 Subject: [PATCH] Format spaces --- examples/unsupervised_learning/MLM/train_mlm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/unsupervised_learning/MLM/train_mlm.py b/examples/unsupervised_learning/MLM/train_mlm.py index 0bc055240..cebc72013 100644 --- a/examples/unsupervised_learning/MLM/train_mlm.py +++ b/examples/unsupervised_learning/MLM/train_mlm.py @@ -27,7 +27,7 @@ use_fp16 = False #Set to True, if your GPU supports FP16 operations max_length = 100 #Max length for a text input do_whole_word_mask = True #If set to true, whole words are masked -mlm_prob = 0.15 #Probability that a word is replaced by a [MASK] token +mlm_prob = 0.15 #Probability that a word is replaced by a [MASK] token # Load the model model = AutoModelForMaskedLM.from_pretrained(model_name)