mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-19 05:54:20 +00:00
Fix CI
This commit is contained in:
parent
8887b28cb8
commit
601de7ed83
@ -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
|
||||
|
||||
|
@ -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 \
|
||||
|
4
.github/scripts/test-ncnn-export.sh
vendored
4
.github/scripts/test-ncnn-export.sh
vendored
@ -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" \
|
||||
|
Loading…
x
Reference in New Issue
Block a user