fix onnx export tests

This commit is contained in:
Fangjun Kuang 2024-09-07 18:59:33 +08:00
parent da6e0a8c68
commit 62bc20a5d2
10 changed files with 16 additions and 2 deletions

View File

@ -25,6 +25,7 @@ popd
log "Export via torch.jit.script()"
./zipformer/export.py \
--use-averaged-model 0 \
--exp-dir $repo/exp \
--tokens $repo/data/lang_bpe_500/tokens.txt \
--epoch 99 \

View File

@ -26,6 +26,8 @@ on:
pull_request:
types: [labeled]
workflow_dispatch:
concurrency:
group: build_doc-${{ github.ref }}
cancel-in-progress: true

View File

@ -42,7 +42,7 @@ concurrency:
jobs:
run_gigaspeech_2023_10_17_zipformer:
if: github.event.label.name == 'zipformer' ||github.event.label.name == 'ready' || github.event.label.name == 'run-decode' || github.event_name == 'push' || github.event_name == 'schedule'
if: github.event.label.name == 'zipformer' ||github.event.label.name == 'ready' || github.event.label.name == 'run-decode' || github.event_name == 'push' || github.event_name == 'schedule' || github.event_name == 'workflow_dispatch'
runs-on: ${{ matrix.os }}
strategy:
matrix:

View File

@ -4,7 +4,6 @@ on:
push:
branches:
- master
- fix-ci
pull_request:
types: [labeled]

View File

@ -23,6 +23,8 @@ on:
pull_request:
types: [labeled]
workflow_dispatch:
concurrency:
group: run_multi-corpora_zipformer-${{ github.ref }}
cancel-in-progress: true

View File

@ -23,6 +23,8 @@ on:
pull_request:
types: [labeled]
workflow_dispatch:
concurrency:
group: run-swbd-conformer_ctc-${{ github.ref }}
cancel-in-progress: true

View File

@ -23,6 +23,8 @@ on:
pull_request:
types: [labeled]
workflow_dispatch:
concurrency:
group: run_wenetspeech_pruned_transducer_stateless2-${{ github.ref }}
cancel-in-progress: true

View File

@ -24,6 +24,8 @@ on:
branches:
- master
workflow_dispatch:
concurrency:
group: style_check-${{ github.ref }}
cancel-in-progress: true

View File

@ -16,6 +16,8 @@ on:
# nightly build at 15:50 UTC time every day
- cron: "50 15 * * *"
workflow_dispatch:
concurrency:
group: test_ncnn_export-${{ github.ref }}
cancel-in-progress: true

View File

@ -16,6 +16,8 @@ on:
# nightly build at 15:50 UTC time every day
- cron: "50 15 * * *"
workflow_dispatch:
concurrency:
group: test_onnx_export-${{ github.ref }}
cancel-in-progress: true