|
#!/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 |
|
|
|
|
