This commit is contained in:
pkufool 2023-06-26 09:11:59 +08:00
parent 601de7ed83
commit 403c3bdb0b
2 changed files with 10 additions and 10 deletions

View File

@ -33,15 +33,15 @@ ln -s pretrained.pt epoch-99.pt
ls -lh *.pt ls -lh *.pt
popd popd
# log "Export to torchscript model" log "Export to torchscript model"
# ./pruned_transducer_stateless7_streaming/export.py \ ./pruned_transducer_stateless7_streaming/export.py \
# --exp-dir $repo/exp \ --exp-dir $repo/exp \
# --use-averaged-model false \ --use-averaged-model false \
# --bpe-model $repo/data/lang_bpe_500/bpe.model \ --bpe-model $repo/data/lang_bpe_500/bpe.model \
# --decode-chunk-len 32 \ --decode-chunk-len 32 \
# --epoch 99 \ --epoch 99 \
# --avg 1 \ --avg 1 \
# --jit 1 --jit 1
ls -lh $repo/exp/*.pt ls -lh $repo/exp/*.pt

View File

@ -275,7 +275,7 @@ def main():
token_table = k2.SymbolTable.from_file(params.tokens) token_table = k2.SymbolTable.from_file(params.tokens)
params.vocab_size = num_tokens(token_table) params.vocab_size = num_tokens(token_table)
params.blank_id = token_table["blk"] params.blank_id = token_table["<blk>"]
assert params.blank_id == 0 assert params.blank_id == 0
logging.info(f"{params}") logging.info(f"{params}")