From 2cc7455b60b0ab0fb3399ca5c1f67a3fa4f86d51 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Thu, 15 Apr 2021 09:06:03 +0000
Subject: [PATCH] fix build vocab

---
 utils/build_vocab.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/build_vocab.py b/utils/build_vocab.py
index dc005656..80576259 100644
--- 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,
-- 
GitLab