do some changes for export.py

This commit is contained in:
luomingshuang 2022-05-06 11:49:08 +08:00
parent ac6d66055a
commit 39f43bc65c

View File

@ -196,7 +196,7 @@ def main():
logging.info(f"device: {device}") logging.info(f"device: {device}")
lexicon = Lexicon(params.lang_dir) lexicon = Lexicon(params.lang_dir)
params.blank_id = 0 params.blank_id = 0
params.vocab_size = max(lexicon.tokens) + 1 params.vocab_size = max(lexicon.tokens) + 1