#!/usr/bin/env bash

set -e

export CUDA_VISIBLE_DEVICES=0

. path.sh
epoch=38
avg=10


./pruned_transducer_stateless2/export.py \
  --exp-dir ./pruned_transducer_stateless2/exp \
  --bpe-model data/lang_bpe_500/bpe.model \
  --epoch $epoch \
  --avg $avg

mv -v ./pruned_transducer_stateless2/exp/pretrained.pt ./pruned_transducer_stateless2/exp/pretrained-epoch-${epoch}-avg-${avg}.pt