This commit is contained in:
pkufool 2023-06-25 16:45:16 +08:00
parent 8887b28cb8
commit 601de7ed83
3 changed files with 12 additions and 13 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

@ -72,8 +72,7 @@ for method in ctc-decoding 1best; do
--use-ctc 1 \
--method $method \
--checkpoint $repo/exp/pretrained.pt \
--tokens $repo/exp/lang_bpe_500/tokens.txt \
--words-file $repo/data/lang_bpe_500/words.txt \
--tokens $repo/data/lang_bpe_500/tokens.txt \
--HLG $repo/data/lang_bpe_500/HLG.pt \
--G $repo/data/lm/G_4_gram.pt \
--words-file $repo/data/lang_bpe_500/words.txt \

View File

@ -195,14 +195,14 @@ git lfs pull --include "data/lang_char_bpe/Linv.pt"
git lfs pull --include "exp/pretrained.pt"
cd exp
ln -s pretrained.pt epoch-99.pt
ln -s pretrained.pt epoch-9999.pt
popd
./pruned_transducer_stateless7_streaming/export-for-ncnn-zh.py \
--lang-dir $repo/data/lang_char_bpe \
--exp-dir $repo/exp \
--use-averaged-model 0 \
--epoch 99 \
--epoch 9999 \
--avg 1 \
--decode-chunk-len 32 \
--num-encoder-layers "2,4,3,2,4" \