CHECKPOINT_PATH=/home/wangrui/projects/SpeechT5/experimental/s2c/checkpoint_28_60000.pt
DATA_ROOT=/home/wangrui/projects/SpeechT5/manifest
SUBSET=test
USER_DIR=/home/wangrui/projects/SpeechT5/SpeechT5/fairseq/examples/speecht5
RESULTS_PATH=/home/wangrui/projects/SpeechT5/experimental/s2c/results

mkdir -p ${RESULTS_PATH}

# fairseq-validate ${DATA_ROOT} \
#   --user-dir ${USER_DIR} \
#   --log-format json \
#   --task speecht5 \
#   --t5-task s2c \
#   --path ${CHECKPOINT_PATH} \
#   --results-path ${RESULTS_PATH} \
#   --batch-size 1 \
#   --max-speech-positions 8000 \
#   --sample-rate 16000 | tee -a ${RESULTS_PATH}/validate-class.txt

python scripts/generate_class.py ${DATA_ROOT} \
  --gen-subset ${SUBSET} \
  --user-dir ${USER_DIR} \
  --log-format json \
  --task speecht5 \
  --t5-task s2c \
  --path ${CHECKPOINT_PATH} \
  --results-path ${RESULTS_PATH} \
  --batch-size 1 \
  --max-speech-positions 8000 \
  --sample-rate 16000 | tee -a ${RESULTS_PATH}/generate-class.txt