mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 18:42:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix blank skip ci test (#1167)

Browse Source 
* Fix for ci

* Fix frame_reducer
This commit is contained in:
Yifan Yang 2023-07-06 23:12:41 +08:00 committed by GitHub
parent 11523c5b89
commit ffe816e2a8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/scripts/run-librispeech-pruned-transducer-stateless7-ctc-bs-2022-12-15.sh → .github/scripts/run-librispeech-pruned-transducer-stateless7-ctc-bs-2023-01-29.sh vendored
View File

@ -21,9 +21,9 @@ tree $repo/
ls -lh $repo/test_wavs/*.wav ls -lh $repo/test_wavs/*.wav
pushd $repo/exp pushd $repo/exp
git lfs pull --include "data/lang_bpe_500/HLG.pt"
git lfs pull --include "data/lang_bpe_500/L.pt" git lfs pull --include "data/lang_bpe_500/L.pt"
git lfs pull --include "data/lang_bpe_500/LG.pt" git lfs pull --include "data/lang_bpe_500/LG.pt"
git lfs pull --include "data/lang_bpe_500/HLG.pt"
git lfs pull --include "data/lang_bpe_500/Linv.pt" git lfs pull --include "data/lang_bpe_500/Linv.pt"
git lfs pull --include "data/lang_bpe_500/bpe.model" git lfs pull --include "data/lang_bpe_500/bpe.model"
git lfs pull --include "exp/cpu_jit.pt" git lfs pull --include "exp/cpu_jit.pt"

8
.github/workflows/run-librispeech-2022-12-15-stateless7-ctc-bs.yml → .github/workflows/run-librispeech-2023-01-29-stateless7-ctc-bs.yml vendored
View File

@ -14,7 +14,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
name: run-librispeech-2022-12-15-stateless7-ctc-bs name: run-librispeech-2023-01-29-stateless7-ctc-bs
# zipformer # zipformer
on: on:
@ -34,7 +34,7 @@ on:
- cron: "50 15 * * *" - cron: "50 15 * * *"
jobs: jobs:
run_librispeech_2022_12_15_zipformer_ctc_bs: run_librispeech_2023_01_29_zipformer_ctc_bs:
if: github.event.label.name == 'run-decode' || github.event.label.name == 'blank-skip' || github.event_name == 'push' || github.event_name == 'schedule' if: github.event.label.name == 'run-decode' || github.event.label.name == 'blank-skip' || github.event_name == 'push' || github.event_name == 'schedule'
runs-on: ${{ matrix.os }} runs-on: ${{ matrix.os }}
strategy: strategy:
@ -124,7 +124,7 @@ jobs:
export PYTHONPATH=~/tmp/kaldifeat/kaldifeat/python:$PYTHONPATH export PYTHONPATH=~/tmp/kaldifeat/kaldifeat/python:$PYTHONPATH
export PYTHONPATH=~/tmp/kaldifeat/build/lib:$PYTHONPATH export PYTHONPATH=~/tmp/kaldifeat/build/lib:$PYTHONPATH
.github/scripts/run-librispeech-pruned-transducer-stateless7-ctc-bs-2022-12-15.sh .github/scripts/run-librispeech-pruned-transducer-stateless7-ctc-bs-2023-01-29.sh
- name: Display decoding results for librispeech pruned_transducer_stateless7_ctc_bs - name: Display decoding results for librispeech pruned_transducer_stateless7_ctc_bs
if: github.event_name == 'schedule' || github.event.label.name == 'run-decode' if: github.event_name == 'schedule' || github.event.label.name == 'run-decode'
@ -159,5 +159,5 @@ jobs:
uses: actions/upload-artifact@v2 uses: actions/upload-artifact@v2
if: github.event_name == 'schedule' || github.event.label.name == 'run-decode' if: github.event_name == 'schedule' || github.event.label.name == 'run-decode'
with: with:
name: torch-${{ matrix.torch }}-python-${{ matrix.python-version }}-ubuntu-latest-cpu-pruned_transducer_stateless7-ctc-bs-2022-12-15 name: torch-${{ matrix.torch }}-python-${{ matrix.python-version }}-ubuntu-latest-cpu-pruned_transducer_stateless7-ctc-bs-2023-01-29
path: egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/exp/ path: egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/exp/

14
egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py
View File

@ -81,20 +81,20 @@ class FrameReducer(nn.Module):
fake_limit_indexes = torch.topk( fake_limit_indexes = torch.topk(
ctc_output[:, :, blank_id], max_limit_len ctc_output[:, :, blank_id], max_limit_len
).indices ).indices
T = ( T_arange = (
torch.arange(max_limit_len) torch.arange(max_limit_len)
.expand_as( .expand_as(
fake_limit_indexes, fake_limit_indexes,
) )
.to(device=x.device) .to(device=x.device)
) )
T = torch.remainder(T, limit_lens.unsqueeze(1)) T_arange = torch.remainder(T_arange, limit_lens.unsqueeze(1))
limit_indexes = torch.gather(fake_limit_indexes, 1, T) limit_indexes = torch.gather(fake_limit_indexes, 1, T_arange)
limit_mask = torch.full_like( limit_mask = torch.full_like(
non_blank_mask, non_blank_mask,
False, 0,
device=x.device, device=x.device,
).scatter_(1, limit_indexes, True) ).scatter_(1, limit_indexes, 1)
non_blank_mask = non_blank_mask | ~limit_mask non_blank_mask = non_blank_mask | ~limit_mask
@ -108,9 +108,9 @@ class FrameReducer(nn.Module):
) )
- out_lens - out_lens
) )
max_pad_len = pad_lens_list.max() max_pad_len = int(pad_lens_list.max())
out = F.pad(x, (0, 0, 0, max_pad_len)) out = F.pad(x, [0, 0, 0, max_pad_len])
valid_pad_mask = ~make_pad_mask(pad_lens_list) valid_pad_mask = ~make_pad_mask(pad_lens_list)
total_valid_mask = torch.concat([non_blank_mask, valid_pad_mask], dim=1) total_valid_mask = torch.concat([non_blank_mask, valid_pad_mask], dim=1)