#!/usr/bin/env bash

export CUDA_VISIBLE_DEVICES=0

# greedy search
./zipformer/decode.py \
  --epoch 30 \
  --avg 9 \
  --exp-dir ./zipformer/exp \
  --max-duration 1000 \
  --decoding-method greedy_search

# modified beam search
./zipformer/decode.py \
  --epoch 30 \
  --avg 9 \
  --exp-dir ./zipformer/exp \
  --max-duration 1000 \
  --decoding-method modified_beam_search \
  --beam-size 4

# fast beam search (one best)
./zipformer/decode.py \
  --epoch 30 \
  --avg 9 \
  --exp-dir ./zipformer/exp \
  --max-duration 1000 \
  --decoding-method fast_beam_search \
  --beam 20.0 \
  --max-contexts 8 \
  --max-states 64