Rename mlm.py to sentence_mlm.py

This script is just for a single sentence MLM.
AntNLP · Jan 19, 2022 · 2e21985 · 2e21985
1 parent 5245189
commit 2e21985
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core-models/mlm.py → core-models/sentence_mlm.py b/core-models/mlm.py → core-models/sentence_mlm.py
@@ -92,4 +92,4 @@ def mask_text(input_file, output_file, mlm_probability=0.15, output_original_tex
 
     # This processing is slow
     # May need ~120 hours for wikipedia-en
-    mask_text(input_file, output_file)
+    mask_text(input_file, output_file)