#!/usr/bin/env bash

. path.sh

export CUDA_VISIBLE_DEVICES=7

set -ex

for epoch in 39 38 37 36 35 34 33 32 32; do
  for avg in 3 4 5 6 7 8 9 10 11; do
    for m in greedy_search; do
      ./pruned_transducer_stateless3/decode.py \
        --exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
        --epoch $epoch \
        --avg $avg \
        --use-averaged-model 1 \
        --max-duration 600 \
        --decoding-method $m \
        --num-encoder-layers 12 \
        --dim-feedforward 2048 \
        --nhead 8 \
        --context-size 1 \
        --encoder-dim 512 \
        --decoder-dim 512 \
        --joiner-dim 512
    done
  done
done