mirror of https://github.com/coqui-ai/TTS.git
37 lines
1.4 KiB
Bash
Executable File
37 lines
1.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -xe
|
|
BASEDIR=$(dirname "$0")
|
|
echo "$BASEDIR"
|
|
|
|
# run training
|
|
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/inputs/test_tacotron_config.json
|
|
# find the training folder
|
|
LATEST_FOLDER=$(ls $BASEDIR/train_outputs/| sort | tail -1)
|
|
echo $LATEST_FOLDER
|
|
# continue the previous training
|
|
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --continue_path $BASEDIR/train_outputs/$LATEST_FOLDER
|
|
# remove all the outputs
|
|
rm -rf $BASEDIR/train_outputs/
|
|
|
|
# run Tacotron bi-directional decoder
|
|
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/inputs/test_tacotron_bd_config.json
|
|
# find the training folder
|
|
LATEST_FOLDER=$(ls $BASEDIR/train_outputs/| sort | tail -1)
|
|
echo $LATEST_FOLDER
|
|
# continue the previous training
|
|
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --continue_path $BASEDIR/train_outputs/$LATEST_FOLDER
|
|
# remove all the outputs
|
|
rm -rf $BASEDIR/train_outputs/
|
|
|
|
# Tacotron2
|
|
# run training
|
|
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/inputs/test_tacotron2_config.json
|
|
# find the training folder
|
|
LATEST_FOLDER=$(ls $BASEDIR/train_outputs/| sort | tail -1)
|
|
echo $LATEST_FOLDER
|
|
# continue the previous training
|
|
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --continue_path $BASEDIR/train_outputs/$LATEST_FOLDER
|
|
# remove all the outputs
|
|
rm -rf $BASEDIR/train_outputs/
|
|
|