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
popd
# log "Export to torchscript model"
# ./pruned_transducer_stateless7_streaming/export.py \
# --exp-dir $repo/exp \
# --use-averaged-model false \
# --bpe-model $repo/data/lang_bpe_500/bpe.model \
# --decode-chunk-len 32 \
# --epoch 99 \
# --avg 1 \
# --jit 1
log "Export to torchscript model"
./pruned_transducer_stateless7_streaming/export.py \
--exp-dir $repo/exp \
--use-averaged-model false \
--bpe-model $repo/data/lang_bpe_500/bpe.model \
--decode-chunk-len 32 \
--epoch 99 \
--avg 1 \
--jit 1
ls -lh $repo/exp/*.pt

View File

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