From 7d44e6b25951ede9b99816039792552a93ec1509 Mon Sep 17 00:00:00 2001 From: jinzr <60612200+JinZr@users.noreply.github.com> Date: Thu, 7 Sep 2023 20:10:40 +0800 Subject: [PATCH] Update run-multi-zh_hans-zipformer.sh --- .../scripts/run-multi-zh_hans-zipformer.sh | 66 ------------------- 1 file changed, 66 deletions(-) diff --git a/.github/scripts/run-multi-zh_hans-zipformer.sh b/.github/scripts/run-multi-zh_hans-zipformer.sh index 6f40b98b6..3a6e1d2f3 100755 --- a/.github/scripts/run-multi-zh_hans-zipformer.sh +++ b/.github/scripts/run-multi-zh_hans-zipformer.sh @@ -26,75 +26,9 @@ pushd $repo/exp ln -s epoch-20.pt epoch-99.pt popd -log "Test exporting to ONNX format" - -./zipformer/export.py \ - --exp-dir $repo/exp \ - --tokens $repo/data/lang_bpe_2000/tokens.txt \ - --epoch 99 \ - --avg 1 \ - --onnx 1 - -log "Export to torchscript model" - -./zipformer/export.py \ - --exp-dir $repo/exp \ - --tokens $repo/data/lang_bpe_2000/tokens.txt \ - --epoch 99 \ - --avg 1 \ - --jit 1 - -./zipformer/export.py \ - --exp-dir $repo/exp \ - --tokens $repo/data/lang_bpe_2000/tokens.txt \ - --epoch 99 \ - --avg 1 \ - --jit-trace 1 - ls -lh $repo/exp/*.onnx ls -lh $repo/exp/*.pt -log "Decode with ONNX models" - -./zipformer/onnx_check.py \ - --jit-filename $repo/exp/cpu_jit.pt \ - --onnx-encoder-filename $repo/exp/encoder.onnx \ - --onnx-decoder-filename $repo/exp/decoder.onnx \ - --onnx-joiner-filename $repo/exp/joiner.onnx \ - --onnx-joiner-encoder-proj-filename $repo/exp/joiner_encoder_proj.onnx \ - --onnx-joiner-decoder-proj-filename $repo/exp/joiner_decoder_proj.onnx - -./zipformer/onnx_pretrained.py \ - --tokens $repo/data/lang_bpe_2000/tokens.txt \ - --encoder-model-filename $repo/exp/encoder.onnx \ - --decoder-model-filename $repo/exp/decoder.onnx \ - --joiner-model-filename $repo/exp/joiner.onnx \ - --joiner-encoder-proj-model-filename $repo/exp/joiner_encoder_proj.onnx \ - --joiner-decoder-proj-model-filename $repo/exp/joiner_decoder_proj.onnx \ - $repo/test_wavs/DEV_T0000000000.wav \ - $repo/test_wavs/DEV_T0000000001.wav \ - $repo/test_wavs/DEV_T0000000002.wav - -log "Decode with models exported by torch.jit.trace()" - -./zipformer/jit_pretrained.py \ - --tokens $repo/data/lang_bpe_2000/tokens.txt \ - --encoder-model-filename $repo/exp/encoder_jit_trace.pt \ - --decoder-model-filename $repo/exp/decoder_jit_trace.pt \ - --joiner-model-filename $repo/exp/joiner_jit_trace.pt \ - $repo/test_wavs/DEV_T0000000000.wav \ - $repo/test_wavs/DEV_T0000000001.wav \ - $repo/test_wavs/DEV_T0000000002.wav - -./zipformer/jit_pretrained.py \ - --tokens $repo/data/lang_bpe_2000/tokens.txt \ - --encoder-model-filename $repo/exp/encoder_jit_script.pt \ - --decoder-model-filename $repo/exp/decoder_jit_script.pt \ - --joiner-model-filename $repo/exp/joiner_jit_script.pt \ - $repo/test_wavs/DEV_T0000000000.wav \ - $repo/test_wavs/DEV_T0000000001.wav \ - $repo/test_wavs/DEV_T0000000002.wav - for sym in 1 2 3; do log "Greedy search with --max-sym-per-frame $sym"