2020-02-13 14:53:45 +08:00
|
|
|
|
|
|
|
# train model
|
2020-02-17 15:53:54 +08:00
|
|
|
CUDA_VISIBLE_DEVICES=0 \
|
2020-02-13 14:53:45 +08:00
|
|
|
python -u synthesis.py \
|
2020-06-19 11:46:10 +08:00
|
|
|
--max_len=400 \
|
|
|
|
--use_gpu=0 \
|
2020-04-09 20:06:04 +08:00
|
|
|
--output='./synthesis' \
|
|
|
|
--config='configs/ljspeech.yaml' \
|
|
|
|
--checkpoint_transformer='./checkpoint/transformer/step-120000' \
|
2020-06-19 11:46:10 +08:00
|
|
|
--vocoder='wavenet' \
|
|
|
|
--config_vocoder='../wavenet/config.yaml' \
|
|
|
|
--checkpoint_vocoder='../wavenet/step-2450000' \
|
|
|
|
#--vocoder='waveflow' \
|
|
|
|
#--config_vocoder='../waveflow/checkpoint/waveflow_res64_ljspeech_ckpt_1.0/waveflow_ljspeech.yaml' \
|
|
|
|
#--checkpoint_vocoder='../waveflow/checkpoint/waveflow_res64_ljspeech_ckpt_1.0/step-3020000' \
|
|
|
|
#--vocoder='cbhg' \
|
|
|
|
#--config_vocoder='configs/ljspeech.yaml' \
|
|
|
|
#--checkpoint_vocoder='checkpoint/cbhg/step-100000' \
|
2020-02-13 14:53:45 +08:00
|
|
|
|
|
|
|
if [ $? -ne 0 ]; then
|
|
|
|
echo "Failed in training!"
|
|
|
|
exit 1
|
|
|
|
fi
|
2020-02-17 15:53:54 +08:00
|
|
|
exit 0
|