fix build vocab

2cc7455b · Hui Zhang · 676ebcdf · 2cc7455b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

utils/build_vocab.py utils/build_vocab.py +1 -1

未找到文件。
--- 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,