Parakeet/examples/transformer_tts/train_vocoder.sh

17 lines
306 B
Bash

# train model
CUDA_VISIBLE_DEVICES=0 \
python -u train_vocoder.py \
--use_gpu=1 \
--data='../../dataset/LJSpeech-1.1' \
--output='./vocoder' \
--config='configs/ljspeech.yaml' \
#--checkpoint='./checkpoint/vocoder/step-100000' \
if [ $? -ne 0 ]; then
echo "Failed in training!"
exit 1
fi
exit 0