-
Notifications
You must be signed in to change notification settings - Fork 306
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add pruned RNN-T for aishell. (#436)
* Add pruned RNN-T for aishell. * support torch script. * Update CI. * Minor fixes. * Add links to sherpa.
- Loading branch information
1 parent
d3daeaf
commit 7100c33
Showing
24 changed files
with
3,055 additions
and
18 deletions.
There are no files selected for viewing
86 changes: 86 additions & 0 deletions
86
.github/scripts/run-aishell-pruned-transducer-stateless3-2022-06-20.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,86 @@ | ||
#!/usr/bin/env bash | ||
|
||
log() { | ||
# This function is from espnet | ||
local fname=${BASH_SOURCE[1]##*/} | ||
echo -e "$(date '+%Y-%m-%d %H:%M:%S') (${fname}:${BASH_LINENO[0]}:${FUNCNAME[1]}) $*" | ||
} | ||
|
||
cd egs/aishell/ASR | ||
|
||
git lfs install | ||
|
||
fbank_url=https://huggingface.co/csukuangfj/aishell-test-dev-manifests | ||
log "Downloading pre-commputed fbank from $fbank_url" | ||
|
||
git clone https://huggingface.co/csukuangfj/aishell-test-dev-manifests | ||
ln -s $PWD/aishell-test-dev-manifests/data . | ||
|
||
log "Downloading pre-trained model from $repo_url" | ||
repo_url=https://huggingface.co/csukuangfj/icefall-aishell-pruned-transducer-stateless3-2022-06-20 | ||
git clone $repo_url | ||
repo=$(basename $repo_url) | ||
|
||
log "Display test files" | ||
tree $repo/ | ||
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 | ||
popd | ||
|
||
for sym in 1 2 3; do | ||
log "Greedy search with --max-sym-per-frame $sym" | ||
|
||
./pruned_transducer_stateless3/pretrained.py \ | ||
--method greedy_search \ | ||
--max-sym-per-frame $sym \ | ||
--checkpoint $repo/exp/pretrained.pt \ | ||
--lang-dir $repo/data/lang_char \ | ||
$repo/test_wavs/BAC009S0764W0121.wav \ | ||
$repo/test_wavs/BAC009S0764W0122.wav \ | ||
$rep/test_wavs/BAC009S0764W0123.wav | ||
done | ||
|
||
for method in modified_beam_search beam_search fast_beam_search; do | ||
log "$method" | ||
|
||
./pruned_transducer_stateless3/pretrained.py \ | ||
--method $method \ | ||
--beam-size 4 \ | ||
--checkpoint $repo/exp/pretrained.pt \ | ||
--lang-dir $repo/data/lang_char \ | ||
$repo/test_wavs/BAC009S0764W0121.wav \ | ||
$repo/test_wavs/BAC009S0764W0122.wav \ | ||
$rep/test_wavs/BAC009S0764W0123.wav | ||
done | ||
|
||
echo "GITHUB_EVENT_NAME: ${GITHUB_EVENT_NAME}" | ||
echo "GITHUB_EVENT_LABEL_NAME: ${GITHUB_EVENT_LABEL_NAME}" | ||
if [[ x"${GITHUB_EVENT_NAME}" == x"schedule" || x"${GITHUB_EVENT_LABEL_NAME}" == x"run-decode" ]]; then | ||
mkdir -p pruned_transducer_stateless3/exp | ||
ln -s $PWD/$repo/exp/pretrained.pt pruned_transducer_stateless3/exp/epoch-999.pt | ||
ln -s $PWD/$repo/data/lang_char data/ | ||
|
||
ls -lh data | ||
ls -lh pruned_transducer_stateless3/exp | ||
|
||
log "Decoding test and dev" | ||
|
||
# use a small value for decoding with CPU | ||
max_duration=100 | ||
|
||
for method in greedy_search fast_beam_search modified_beam_search; do | ||
log "Decoding with $method" | ||
|
||
./pruned_transducer_stateless3/decode.py \ | ||
--decoding-method $method \ | ||
--epoch 999 \ | ||
--avg 1 \ | ||
--max-duration $max_duration \ | ||
--exp-dir pruned_transducer_stateless3/exp | ||
done | ||
|
||
rm pruned_transducer_stateless3/exp/*.pt | ||
fi |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,119 @@ | ||
# Copyright 2022 Fangjun Kuang ([email protected]) | ||
|
||
# See ../../LICENSE for clarification regarding multiple authors | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
name: run-aishell-2022-06-20 | ||
# pruned RNN-T + reworked model with random combiner | ||
# https://huggingface.co/csukuangfj/icefall-aishell-pruned-transducer-stateless3-2022-06-20 | ||
|
||
on: | ||
push: | ||
branches: | ||
- master | ||
pull_request: | ||
types: [labeled] | ||
|
||
schedule: | ||
# minute (0-59) | ||
# hour (0-23) | ||
# day of the month (1-31) | ||
# month (1-12) | ||
# day of the week (0-6) | ||
# nightly build at 15:50 UTC time every day | ||
- cron: "50 15 * * *" | ||
|
||
jobs: | ||
run_aishell_2022_06_20: | ||
if: github.event.label.name == 'ready' || github.event.label.name == 'run-decode' || github.event_name == 'push' || github.event_name == 'schedule' | ||
runs-on: ${{ matrix.os }} | ||
strategy: | ||
matrix: | ||
os: [ubuntu-18.04] | ||
python-version: [3.7, 3.8, 3.9] | ||
|
||
fail-fast: false | ||
|
||
steps: | ||
- uses: actions/checkout@v2 | ||
with: | ||
fetch-depth: 0 | ||
|
||
- name: Setup Python ${{ matrix.python-version }} | ||
uses: actions/setup-python@v2 | ||
with: | ||
python-version: ${{ matrix.python-version }} | ||
cache: 'pip' | ||
cache-dependency-path: '**/requirements-ci.txt' | ||
|
||
- name: Install Python dependencies | ||
run: | | ||
grep -v '^#' ./requirements-ci.txt | xargs -n 1 -L 1 pip install | ||
pip uninstall -y protobuf | ||
pip install --no-binary protobuf protobuf | ||
- name: Cache kaldifeat | ||
id: my-cache | ||
uses: actions/cache@v2 | ||
with: | ||
path: | | ||
~/tmp/kaldifeat | ||
key: cache-tmp-${{ matrix.python-version }} | ||
|
||
- name: Install kaldifeat | ||
if: steps.my-cache.outputs.cache-hit != 'true' | ||
shell: bash | ||
run: | | ||
.github/scripts/install-kaldifeat.sh | ||
- name: Inference with pre-trained model | ||
shell: bash | ||
env: | ||
GITHUB_EVENT_NAME: ${{ github.event_name }} | ||
GITHUB_EVENT_LABEL_NAME: ${{ github.event.label.name }} | ||
run: | | ||
sudo apt-get -qq install git-lfs tree sox | ||
export PYTHONPATH=$PWD:$PYTHONPATH | ||
export PYTHONPATH=~/tmp/kaldifeat/kaldifeat/python:$PYTHONPATH | ||
export PYTHONPATH=~/tmp/kaldifeat/build/lib:$PYTHONPATH | ||
.github/scripts/run-aishell-pruned-transducer-stateless3-2022-06-20.sh | ||
- name: Display decoding results for aishell pruned_transducer_stateless3 | ||
if: github.event_name == 'schedule' || github.event.label.name == 'run-decode' | ||
shell: bash | ||
run: | | ||
cd egs/aishell/ASR/ | ||
tree ./pruned_transducer_stateless3/exp | ||
cd pruned_transducer_stateless3 | ||
echo "results for pruned_transducer_stateless3" | ||
echo "===greedy search===" | ||
find exp/greedy_search -name "log-*" -exec grep -n --color "best for test" {} + | sort -n -k2 | ||
find exp/greedy_search -name "log-*" -exec grep -n --color "best for dev" {} + | sort -n -k2 | ||
echo "===fast_beam_search===" | ||
find exp/fast_beam_search -name "log-*" -exec grep -n --color "best for test" {} + | sort -n -k2 | ||
find exp/fast_beam_search -name "log-*" -exec grep -n --color "best for dev" {} + | sort -n -k2 | ||
echo "===modified beam search===" | ||
find exp/modified_beam_search -name "log-*" -exec grep -n --color "best for test" {} + | sort -n -k2 | ||
find exp/modified_beam_search -name "log-*" -exec grep -n --color "best for dev" {} + | sort -n -k2 | ||
- name: Upload decoding results for aishell pruned_transducer_stateless3 | ||
uses: actions/upload-artifact@v2 | ||
if: github.event_name == 'schedule' || github.event.label.name == 'run-decode' | ||
with: | ||
name: aishell-torch-${{ matrix.torch }}-python-${{ matrix.python-version }}-ubuntu-18.04-cpu-pruned_transducer_stateless3-2022-06-20 | ||
path: egs/aishell/ASR/pruned_transducer_stateless3/exp/ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
1 change: 1 addition & 0 deletions
1
egs/aishell/ASR/pruned_transducer_stateless3/aidatatang_200zh.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../transducer_stateless_modified-2/aidatatang_200zh.py |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../transducer_stateless_modified-2/aishell.py |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../transducer_stateless_modified-2/asr_datamodule.py |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../../../librispeech/ASR/pruned_transducer_stateless2/beam_search.py |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
../../../librispeech/ASR/pruned_transducer_stateless5/conformer.py |
Oops, something went wrong.