Export speaker verification models from NeMo to ONNX (#526)

This commit is contained in:
Fangjun Kuang
2024-01-13 19:49:45 +08:00
committed by GitHub
parent afc81ec122
commit 68a525a024
10 changed files with 448 additions and 28 deletions

View File

@@ -0,0 +1,45 @@
name: export-nemo-speaker-verification-to-onnx
on:
workflow_dispatch:
concurrency:
group: export-nemo-speaker-verification-to-onnx-${{ github.ref }}
cancel-in-progress: true
jobs:
export-nemo-speaker-verification-to-onnx:
if: github.repository_owner == 'k2-fsa' || github.repository_owner == 'csukuangfj'
name: export nemo speaker verification models to ONNX
runs-on: ${{ matrix.os }}
strategy:
fail-fast: false
matrix:
os: [ubuntu-latest]
python-version: ["3.10"]
steps:
- uses: actions/checkout@v4
- name: Setup Python ${{ matrix.python-version }}
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}
- name: Run
shell: bash
run: |
cd scripts/nemo/speaker-verification
./run.sh
mv -v *.onnx ../../..
- name: Release
uses: svenstaro/upload-release-action@v2
with:
file_glob: true
file: ./*.onnx
overwrite: true
repo_name: k2-fsa/sherpa-onnx
repo_token: ${{ secrets.UPLOAD_GH_SHERPA_ONNX_TOKEN }}
tag: speaker-recongition-models