#!/usr/bin/env bash

set -e

export CUDA_VISIBLE_DEVICES=0

. path.sh
epoch=39
avg=13
use_averaged_model=true


./pruned_transducer_stateless5/export.py \
  --exp-dir ./pruned_transducer_stateless5/exp-L \
  --bpe-model data/lang_bpe_500/bpe.model \
  --epoch $epoch \
  --avg $avg \
  --num-encoder-layers 18 \
  --dim-feedforward 2048 \
  --nhead 8 \
  --encoder-dim 512 \
  --decoder-dim 512 \
  --joiner-dim 512 \
  --use-averaged-model $use_averaged_model

mv -v ./pruned_transducer_stateless5/exp-L/pretrained.pt ./pruned_transducer_stateless5/exp-L/pretrained-epoch-${epoch}-avg-${avg}-use-averaged-model-${use_averaged_model}.pt