#!/usr/bin/env bash

set -x

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

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

export CUDA_VISIBLE_DEVICES=""

./pruned_transducer_stateless5/export.py \
        --streaming-model 1 \
        --causal-convolution 1 \
        --num-encoder-layers 18 \
        --dim-feedforward 2048 \
        --nhead 8 \
        --encoder-dim 512 \
        --decoder-dim 512 \
        --joiner-dim 512 \
        --epoch 25 \
        --avg 5 \
        --exp-dir ./pruned_transducer_stateless5/exp-L-nolinear

./pruned_transducer_stateless5/export.py \
        --streaming-model 1 \
        --causal-convolution 1 \
        --num-encoder-layers 18 \
        --dim-feedforward 2048 \
        --nhead 8 \
        --encoder-dim 512 \
        --decoder-dim 512 \
        --joiner-dim 512 \
        --epoch 25 \
        --avg 5 \
        --exp-dir ./pruned_transducer_stateless5/exp-L-nolinear \
        --jit 1