#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES=0

set -ex

epoch=39
avg=7

./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

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