mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-07 08:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
marcoyang 2023-10-11 14:40:39 +08:00
parent 9416dde02f
commit b21e4da943
2 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/libriheavy/ASR/RESULTS.md
View File

@ -103,7 +103,7 @@ use_context_list=True
python ./zipformer_prompt_asr/train_bert_encoder.py \
--world-size 4 \
--start-epoch 1 \
--num-epochs 60 \
--num-epochs 50 \
--exp-dir ./zipformer_prompt_asr/exp \
--use-fp16 True \
--memory-dropout-rate $memory_dropout_rate \

20
egs/libriheavy/ASR/zipformer_prompt_asr/export_PromptASR.py
View File

@ -24,20 +24,20 @@ Export `model.state_dict()`
- For non-streaming model:
./zipformer/export_PromptASR.py \
--exp-dir ./zipformer/exp \
--tokens data/lang_bpe_500/tokens.txt \
--epoch 30 \
--avg 9
./zipformer_prompt_asr/export_PromptASR.py \
--exp-dir ./zipformer_prompt_asr/exp \
--tokens data/lang_bpe_500_fallback_coverage_0.99/tokens.txt \
--epoch 50 \
--avg 10
- For streaming model:
./zipformer/export_PromptASR.py \
--exp-dir ./zipformer/exp \
./zipformer_prompt_asr/export_PromptASR.py \
--exp-dir ./zipformer_prompt_asr/exp \
--causal 1 \
--tokens data/lang_bpe_500/tokens.txt \
--epoch 30 \
--avg 9
--tokens data/lang_bpe_500_fallback_coverage_0.99/tokens.txt \
--epoch 50 \
--avg 10
It will generate a file `pretrained.pt` in the given `exp_dir`. You can later
load it by `icefall.checkpoint.load_checkpoint()`.