diff --git a/egs/gigaspeech/KWS/run.sh b/egs/gigaspeech/KWS/run.sh index bd562ce1c..303abd718 100755 --- a/egs/gigaspeech/KWS/run.sh +++ b/egs/gigaspeech/KWS/run.sh @@ -90,7 +90,7 @@ if [ $stage -le 2 ] && [ $stop_stage -ge 2 ]; then --encoder-dim 128,128,128,128,128,128 \ --encoder-unmasked-dim 128,128,128,128,128,128 - python ./zipformer/export_onnx_streaming.py \ + python ./zipformer/export-onnx-streaming.py \ --exp-dir zipformer/exp \ --tokens data/lang_bpe_500/tokens.txt \ --epoch 12 \ @@ -184,7 +184,7 @@ if [ $stage -le 5 ] && [ $stop_stage -ge 5 ]; then --encoder-dim 128,128,128,128,128,128 \ --encoder-unmasked-dim 128,128,128,128,128,128 - python ./zipformer/export_onnx_streaming.py \ + python ./zipformer/export-onnx-streaming.py \ --exp-dir zipformer/exp_finetune \ --tokens data/lang_bpe_500/tokens.txt \ --epoch 10 \ diff --git a/egs/wenetspeech/KWS/run.sh b/egs/wenetspeech/KWS/run.sh index 232ee039a..8472b8531 100755 --- a/egs/wenetspeech/KWS/run.sh +++ b/egs/wenetspeech/KWS/run.sh @@ -91,7 +91,7 @@ if [ $stage -le 2 ] && [ $stop_stage -ge 2 ]; then --encoder-dim 128,128,128,128,128,128 \ --encoder-unmasked-dim 128,128,128,128,128,128 - python ./zipformer/export_onnx_streaming.py \ + python ./zipformer/export-onnx-streaming.py \ --exp-dir zipformer/exp \ --tokens data/lang_partial_tone/tokens.txt \ --epoch 18 \ @@ -187,7 +187,7 @@ if [ $stage -le 5 ] && [ $stop_stage -ge 5 ]; then --encoder-dim 128,128,128,128,128,128 \ --encoder-unmasked-dim 128,128,128,128,128,128 - python ./zipformer/export_onnx_streaming.py \ + python ./zipformer/export-onnx-streaming.py \ --exp-dir zipformer/exp_finetune \ --tokens data/lang_partial_tone/tokens.txt \ --epoch 10 \