#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES="7"

. path.sh
set -ex

# epoch=1
# iter=2
mode=1
use_averaged_model=1

epoch=39
iter=116000
avg_list="11 12 13 14 15"

if [ $mode -eq 1 ]; then
  for avg in $avg_list; do
    ./pruned_transducer_stateless5/decode.py \
      --epoch $epoch \
      --avg $avg \
      --exp-dir ./pruned_transducer_stateless5/exp-L \
      --max-duration 600 \
      --decoding-method fast_beam_search \
      --max-sym-per-frame 1 \
      --num-encoder-layers 18 \
      --dim-feedforward 2048 \
      --nhead 8 \
      --encoder-dim 512 \
      --decoder-dim 512 \
      --joiner-dim 512 \
      --use-averaged-model $use_averaged_model
  done
elif [ $mode -eq 2 ]; then
  for avg in $avg_list; do
    ./pruned_transducer_stateless5/decode.py \
      --iter $iter \
      --avg $avg \
      --exp-dir ./pruned_transducer_stateless5/exp-L \
      --max-duration 600 \
      --decoding-method greedy_search \
      --max-sym-per-frame 1 \
      --num-encoder-layers 18 \
      --dim-feedforward 2048 \
      --nhead 8 \
      --encoder-dim 512 \
      --decoder-dim 512 \
      --joiner-dim 512 \
      --use-averaged-model $use_averaged_model
  done
else
  echo "Unsupported mode $mode"
fi