#!/usr/bin/env bash

set -x

K2_ROOT=/ceph-hw/kangwei/code/k2_release/k2
ICEFALL=/ceph-hw/kangwei/code2/icefall_bbpe

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

export CUDA_VISIBLE_DEVICES=""

./pruned_transducer_stateless7_bbpe/export.py \
        --epoch 35 \
        --avg 26 \
        --bpe-model data/lang_bbpe_500/bbpe.model \
        --exp-dir ./pruned_transducer_stateless7_bbpe/exp

./pruned_transducer_stateless7_bbpe/export.py \
        --epoch 35 \
        --avg 26 \
        --bpe-model data/lang_bbpe_500/bbpe.model \
        --exp-dir ./pruned_transducer_stateless7_bbpe/exp
        --jit 1