scripts/export.sh

#!/usr/bin/env bash

set -x

K2_ROOT=/ceph-hw/kangwei/code/k2_release/k2
ICEFALL=/star-kw/kangwei/code/icefall_wenetspeech

export PYTHONPATH=$K2_ROOT/k2/python:$PYTHONPATH
export PYTHONPATH=$K2_ROOT/build/lib:$PYTHONPATH
export PYTHONPATH=$ICEFALL:$PYTHONPATH

export CUDA_VISIBLE_DEVICES=""
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python


./zipformer/export-onnx-streaming.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --chunk-size 16 \
  --left-context-frames 128 \
  --causal 1

./zipformer/export-onnx-streaming.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --chunk-size 16 \
  --left-context-frames 256 \
  --causal 1

./zipformer/export-onnx-streaming.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --chunk-size 32 \
  --left-context-frames 256 \
  --causal 1

./zipformer/export.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --causal 1 \
  --avg 4

./zipformer/export.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --causal 1 \
  --chunk-size 16 \
  --left-context-frames 128 \
  --jit 1

./zipformer/export.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --causal 1 \
  --chunk-size 16 \
  --left-context-frames 256 \
  --jit 1

./zipformer/export.py \
  --exp-dir zipformer/exp_L_causal_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --causal 1 \
  --chunk-size 32 \
  --left-context-frames 256 \
  --jit 1