run_facodec.sh

#!/bin/bash
# Apache 2.0.

# args
input_feature_dim=768
dropout_p=0.6

stage=1
stop_stage=9

# prepare train data
if [ ${stage} -le 1 ] && [ ${stop_stage} -ge 1 ]; then
  echo "Stage 1"
  python facodec/infer_timbre_emb.py
fi

# prepare test data
if [ ${stage} -le 2 ] && [ ${stop_stage} -ge 2 ]; then
  echo "Stage 2"
  python facodec/infer_timbre_emb_test.py
fi

# train model
if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
  echo "Stage 3"
  python train_wespeaker.py \
  --config_model_name "train_facodec" \
  --input_feature_dim $input_feature_dim \
  --dropout_p $dropout_p \
  --add_ab
fi

# infer validate set
if [ ${stage} -le 4 ] && [ ${stop_stage} -ge 4 ]; then
  echo "Stage 4"
  python infer_wespeaker.py \
  --config_model_name "infer_facodec" \
  --input_feature_dim $input_feature_dim \
  --dropout_p $dropout_p \
  --add_ab
fi

# acc_eer validate set
if [ ${stage} -le 5 ] && [ ${stop_stage} -ge 5 ]; then
  echo "Stage 5"
  python acc_eer.py \
  --config_path "configs/baseline.json" \
  --config_model "acc_eer_facodec"
fi

# infer seen test set
if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then
  echo "Stage 6"
  python infer_wespeaker_test.py \
  --config_name "seen_infer_facodec" \
  --input_feature_dim $input_feature_dim \
  --dropout_p $dropout_p \
  --add_ab
fi

# infer unseen test set
if [ ${stage} -le 7 ] && [ ${stop_stage} -ge 7 ]; then
  echo "Stage 7"
  python infer_wespeaker_test.py \
  --config_name "unseen_infer_facodec" \
  --input_feature_dim $input_feature_dim \
  --dropout_p $dropout_p \
  --add_ab
fi

# acc_eer seen test set
if [ ${stage} -le 8 ] && [ ${stop_stage} -ge 8 ]; then
  echo "Stage 8"
  python acc_eer.py \
  --config_path "configs/baseline.json" \
  --config_model "acc_eer_seen_infer_facodec"
fi

# acc_eer unseen test set
if [ ${stage} -le 9 ] && [ ${stop_stage} -ge 9 ]; then
  echo "Stage 9"
  python acc_eer.py \
  --config_path "configs/baseline.json" \
  --config_model "acc_eer_unseen_infer_facodec"
fi