fix build vocab

5 years ago · 2cc7455b60
parent 676ebcdff5
commit 2cc7455b60
1 changed files with 1 additions and 1 deletions
--- a/utils/build_vocab.py
+++ b/utils/build_vocab.py
@ -83,7 +83,7 @@ def main():
        # train
        spm.SentencePieceTrainer.Train(
            input=fp.name,
-            vocab_size=args.vocab_size,
+            vocab_size=args.spm_vocab_size,
            model_type=args.spm_mode,
            model_prefix=args.spm_model_prefix,
            input_sentence_size=100000000,