run-attn-conformer.sh

#!/usr/bin/env bash
# Attention-model 

stage=1

. ./cmd.sh
. ./path.sh
. parse_options.sh

# you might not want to do this for interactive shells.
set -e

# RUN run.sh first, this builds on those shards


if [ $stage -le 1 ]; then
  sbatch local/attention/run_training.sh --hparams hyperparams/attention/W2V2-A.yaml --py_script local/attention/sb_train_attn_w2v2.py
  echo "Submitted training job, exiting"
  exit
fi

if [ $stage -le 3 ]; then
  sbatch local/attention/run_training.sh --py_script local/attention/sb_train_attn_mwer_w2v2.py --hparams hyperparams/attention/mwer/W2V2-a.yaml
  echo "Submitted training job, exiting"
  exit
fi