mirrors/icefall
Archived
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

do some changes for export.py

Browse Source
This commit is contained in:
luomingshuang 2022-07-28 10:22:26 +08:00
parent 23be42a8c9
commit 44ec59fc1f
1 changed files with 39 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
egs/wenetspeech/ASR/pruned_transducer_stateless5/export.py
View File

@ -17,26 +17,54 @@
# This script converts several saved checkpoints # This script converts several saved checkpoints
# to a single one using model averaging. # to a single one using model averaging.
""" """
Usage: Usage for offline:
./pruned_transducer_stateless2/export.py \ ./pruned_transducer_stateless5/export.py \
--exp-dir ./pruned_transducer_stateless2/exp \ --exp-dir ./pruned_transducer_stateless5/exp_L_offline \
--lang-dir data/lang_char \ --lang-dir data/lang_char \
--epoch 10 \ --epoch 4 \
--avg 2 --avg 1
It will generate a file exp_dir/pretrained.pt It will generate a file exp_dir/pretrained.pt for offline ASR.
To use the generated file with `pruned_transducer_stateless2/decode.py`, ./pruned_transducer_stateless5/export.py \
--exp-dir ./pruned_transducer_stateless5/exp_L_offline \
--lang-dir data/lang_char \
--epoch 4 \
--avg 1 \
--jit True
It will generate a file exp_dir/cpu_jit.pt for offline ASR.
Usage for streaming:
./pruned_transducer_stateless5/export.py \
--exp-dir ./pruned_transducer_stateless5/exp_L_streaming \
--lang-dir data/lang_char \
--epoch 7 \
--avg 1
It will generate a file exp_dir/pretrained.pt for streaming ASR.
./pruned_transducer_stateless5/export.py \
--exp-dir ./pruned_transducer_stateless5/exp_L_streaming \
--lang-dir data/lang_char \
--epoch 7 \
--avg 1 \
--jit True
It will generate a file exp_dir/cpu_jit.pt for streaming ASR.
To use the generated file with `pruned_transducer_stateless5/decode.py`,
you can do: you can do:
cd /path/to/exp_dir cd /path/to/exp_dir
ln -s pretrained.pt epoch-9999.pt ln -s pretrained.pt epoch-9999.pt
cd /path/to/egs/wenetspeech/ASR cd /path/to/egs/wenetspeech/ASR
./pruned_transducer_stateless2/decode.py \ ./pruned_transducer_stateless5/decode.py \
--exp-dir ./pruned_transducer_stateless2/exp \ --exp-dir ./pruned_transducer_stateless5/exp \
--epoch 10 \ --epoch 4 \
--avg 2 \ --avg 1 \
--decoding-method greedy_search \
--max-duration 100 \ --max-duration 100 \
--lang-dir data/lang_char --lang-dir data/lang_char
""" """