Set batch size to 1 for more streaming ASR models (#1280)

This commit is contained in:
Fangjun Kuang
2024-08-23 11:06:55 +08:00
committed by GitHub
parent c61423ec5a
commit fb09f8fae3
15 changed files with 782 additions and 38 deletions

View File

@@ -7,7 +7,6 @@ on:
workflow_dispatch:
concurrency:
group: mobile-asr-models-${{ github.ref }}
cancel-in-progress: true
@@ -16,11 +15,14 @@ jobs:
mobile-asr-models:
if: github.repository_owner == 'k2-fsa' || github.repository_owner == 'csukuangfj' || github.repository_owner == 'csu-fangjun'
runs-on: ${{ matrix.os }}
name: ${{ matrix.index }}/${{ matrix.total }}
strategy:
fail-fast: false
matrix:
os: [ubuntu-latest]
python-version: ["3.8"]
total: ["11"]
index: ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "10"]
steps:
- uses: actions/checkout@v4
@@ -33,7 +35,20 @@ jobs:
- name: Install dependencies
shell: bash
run: |
python3 -m pip install onnxruntime==1.16.3 onnx==1.15.0
python3 -m pip install onnxruntime==1.16.3 onnx==1.15.0 jinja2
- name: Generate build script
shell: bash
run: |
cd scripts/mobile-asr-models
total=${{ matrix.total }}
index=${{ matrix.index }}
./generate-asr.py --total $total --index $index
chmod +x run2.sh
mv run2.sh run.sh
ls -lh
- name: Run
shell: bash