From b21e4da9434bc01da97ed63f3c2b7bcc54ff3b99 Mon Sep 17 00:00:00 2001 From: marcoyang Date: Wed, 11 Oct 2023 14:40:39 +0800 Subject: [PATCH] minor fixes --- egs/libriheavy/ASR/RESULTS.md | 2 +- .../zipformer_prompt_asr/export_PromptASR.py | 20 +++++++++---------- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/egs/libriheavy/ASR/RESULTS.md b/egs/libriheavy/ASR/RESULTS.md index 79fd898bf..4fbedad98 100644 --- a/egs/libriheavy/ASR/RESULTS.md +++ b/egs/libriheavy/ASR/RESULTS.md @@ -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 \ diff --git a/egs/libriheavy/ASR/zipformer_prompt_asr/export_PromptASR.py b/egs/libriheavy/ASR/zipformer_prompt_asr/export_PromptASR.py index b0cb1c46f..e0bc556a8 100644 --- a/egs/libriheavy/ASR/zipformer_prompt_asr/export_PromptASR.py +++ b/egs/libriheavy/ASR/zipformer_prompt_asr/export_PromptASR.py @@ -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()`.