From c5e697baa2b68bf4f1409d62fc0110b6d878683d Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Wed, 6 Jul 2022 14:00:57 +0800 Subject: [PATCH 1/3] Add multi_quantization to requirements.txt --- requirements-ci.txt | 1 + requirements.txt | 1 + 2 files changed, 2 insertions(+) diff --git a/requirements-ci.txt b/requirements-ci.txt index 4f507285b5..fc17b123a8 100644 --- a/requirements-ci.txt +++ b/requirements-ci.txt @@ -19,3 +19,4 @@ kaldialign==0.2 sentencepiece==0.1.96 tensorboard==2.8.0 typeguard==2.13.3 +multi_quantization diff --git a/requirements.txt b/requirements.txt index 4eaa86a678..90b1dac698 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,3 +3,4 @@ kaldialign sentencepiece>=0.1.96 tensorboard typeguard +multi_quantization From a3d0b4b295e7733634bd556b3d4abe56412c7955 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Wed, 6 Jul 2022 16:08:38 +0800 Subject: [PATCH 2/3] Minor fixes. --- .../run-aishell-pruned-transducer-stateless3-2022-06-20.sh | 2 +- .../run-librispeech-pruned-transducer-stateless5-2022-05-13.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/scripts/run-aishell-pruned-transducer-stateless3-2022-06-20.sh b/.github/scripts/run-aishell-pruned-transducer-stateless3-2022-06-20.sh index cf35f711b5..631707ad91 100755 --- a/.github/scripts/run-aishell-pruned-transducer-stateless3-2022-06-20.sh +++ b/.github/scripts/run-aishell-pruned-transducer-stateless3-2022-06-20.sh @@ -27,7 +27,7 @@ soxi $repo/test_wavs/*.wav ls -lh $repo/test_wavs/*.wav pushd $repo/exp -ln -s pretrained-epoch-29-avg-5-torch-1.10.pt pretrained.pt +ln -s pretrained-epoch-29-avg-5-torch-1.10.0.pt pretrained.pt popd for sym in 1 2 3; do diff --git a/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh b/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh index 61210ac6e6..e71f6ea280 100755 --- a/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh +++ b/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh @@ -37,7 +37,7 @@ for sym in 1 2 3; do --nhead 8 \ --encoder-dim 512 \ --decoder-dim 512 \ - --joiner-dim 512 + --joiner-dim 512 \ $repo/test_wavs/1089-134686-0001.wav \ $repo/test_wavs/1221-135766-0001.wav \ $repo/test_wavs/1221-135766-0002.wav From a88f77877a54d13cd148ed7aba0a7ef77b2243a1 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Wed, 6 Jul 2022 20:29:06 +0800 Subject: [PATCH 3/3] more fixes --- .../run-librispeech-pruned-transducer-stateless5-2022-05-13.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh b/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh index e71f6ea280..c893bc45ae 100755 --- a/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh +++ b/.github/scripts/run-librispeech-pruned-transducer-stateless5-2022-05-13.sh @@ -82,6 +82,7 @@ if [[ x"${GITHUB_EVENT_NAME}" == x"schedule" || x"${GITHUB_EVENT_LABEL_NAME}" == ./pruned_transducer_stateless5/decode.py \ --decoding-method $method \ + --use-averaged-model 0 \ --epoch 999 \ --avg 1 \ --max-duration $max_duration \