#!/usr/bin/env bash
. path.sh

set -ex

for jit in 0 1; do
  ./pruned_transducer_stateless3/export.py \
    --jit $jit \
    --context-size 1 \
    --exp-dir ./pruned_transducer_stateless3/exp-context-size-1 \
    --epoch 29 \
    --avg 5 \
    --num-encoder-layers 12 \
    --dim-feedforward 2048 \
    --nhead 8 \
    --encoder-dim 512 \
    --context-size 1 \
    --decoder-dim 512 \
    --joiner-dim 512
done