#!/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.py \
  --exp-dir zipformer/exp_L_context_2 \
  --tokens data/lang_char/tokens.txt \
  --epoch 12 \
  --avg 4 \
  --causal 0

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

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