#!/usr/bin/env bash

. path.sh

export CUDA_VISIBLE_DEVICES=""

set -ex

epoch=15
avg=3
jit=1

dir=./conv_emformer_transducer_stateless2

./conv_emformer_transducer_stateless2/export.py \
  --exp-dir $dir/exp-L-small \
  --lang-dir ./data/lang_char \
  --epoch $epoch \
  --avg $avg \
  --use-averaged-model 1 \
  \
  --encoder-dim 256 \
  --nhead 4 \
  --dim-feedforward 1024 \
  --num-encoder-layers 5 \
  --chunk-length 32 \
  --cnn-module-kernel 31 \
  --left-context-length 32 \
  --right-context-length 8 \
  --memory-size 32 \
  --jit $jit

if [ $jit -eq 0 ]; then
  pushd $dir/exp-L-small
  mv pretrained.pt pretrained-epoch-$epoch-avg-$avg.pt
  popd
fi

if [ $jit -eq 1 ]; then
  pushd $dir/exp-L-small
  mv cpu_jit.pt cpu_jit-epoch-$epoch-avg-$avg.pt
  popd
fi