#!/usr/bin/env bash


export CUDA_VISIBLE_DEVICES=0

. path.sh

set -ex

iter=
epoch=39
avg=6
use_averaged_model=1
jit=0


./pruned_stateless_emformer_rnnt2/export.py \
  --exp-dir ./pruned_stateless_emformer_rnnt2/exp-full \
  --bpe-model data/lang_bpe_500/bpe.model \
  --epoch $epoch \
  --avg $avg \
  --jit $jit \
  --use-averaged-model $use_averaged_model

if [ $jit -eq 1 ]; then
  mv -v ./pruned_stateless_emformer_rnnt2/exp-full/cpu_jit.pt ./pruned_stateless_emformer_rnnt2/exp-full/cpu_jit-epoch-${epoch}-avg-${avg}-use-averaged-model-${use_averaged_model}.pt
else
  mv -v ./pruned_stateless_emformer_rnnt2/exp-full/pretrained.pt ./pruned_stateless_emformer_rnnt2/exp-full/pretrained-epoch-${epoch}-avg-${avg}-use-averaged-model-${use_averaged_model}.pt
fi