TTS/tests/bash_tests/test_tacotron_train.sh

37 lines
1.4 KiB
Bash
Raw Normal View History

2020-07-16 13:05:36 +00:00
#!/usr/bin/env bash
2020-11-12 15:33:29 +00:00
set -xe
2020-07-16 13:05:36 +00:00
BASEDIR=$(dirname "$0")
echo "$BASEDIR"
2021-03-15 10:28:06 +00:00
2020-07-16 13:05:36 +00:00
# run training
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/../inputs/test_tacotron_config.json
2020-07-16 13:05:36 +00:00
# find the training folder
LATEST_FOLDER=$(ls $BASEDIR/../train_outputs/| sort | tail -1)
2020-07-16 13:05:36 +00:00
echo $LATEST_FOLDER
# continue the previous training
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --continue_path $BASEDIR/../train_outputs/$LATEST_FOLDER
2020-07-16 13:05:36 +00:00
# remove all the outputs
rm -rf $BASEDIR/../train_outputs/
2021-03-11 17:14:53 +00:00
2021-03-15 10:28:06 +00:00
# run Tacotron bi-directional decoder
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/../inputs/test_tacotron_bd_config.json
2021-03-15 10:28:06 +00:00
# find the training folder
LATEST_FOLDER=$(ls $BASEDIR/../train_outputs/| sort | tail -1)
2021-03-15 10:28:06 +00:00
echo $LATEST_FOLDER
# continue the previous training
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --continue_path $BASEDIR/../train_outputs/$LATEST_FOLDER
2021-03-15 10:28:06 +00:00
# remove all the outputs
rm -rf $BASEDIR/../train_outputs/
2021-03-15 10:28:06 +00:00
2021-03-11 17:14:53 +00:00
# Tacotron2
# run training
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --config_path $BASEDIR/../inputs/test_tacotron2_config.json
2021-03-11 17:14:53 +00:00
# find the training folder
LATEST_FOLDER=$(ls $BASEDIR/../train_outputs/| sort | tail -1)
2021-03-11 17:14:53 +00:00
echo $LATEST_FOLDER
# continue the previous training
CUDA_VISIBLE_DEVICES="" python TTS/bin/train_tacotron.py --continue_path $BASEDIR/../train_outputs/$LATEST_FOLDER
2021-03-11 17:14:53 +00:00
# remove all the outputs
rm -rf $BASEDIR/../train_outputs/
2021-03-11 17:14:53 +00:00