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-03-05 15:08:12 +08:00
|
|
|
--max_len=600 \
|
2020-02-13 14:53:45 +08:00
|
|
|
--transformer_step=160000 \
|
2020-03-05 15:08:12 +08:00
|
|
|
--vocoder_step=90000 \
|
|
|
|
|
--use_gpu=1 \
|
2020-02-13 14:53:45 +08:00
|
|
|
--checkpoint_path='./checkpoint' \
|
|
|
|
|
--log_dir='./log' \
|
|
|
|
|
--sample_path='./sample' \
|
|
|
|
|
--config_path='config/synthesis.yaml' \
|
|
|
|
|
|
|
|
|
|
if [ $? -ne 0 ]; then
|
|
|
|
|
echo "Failed in training!"
|
|
|
|
|
exit 1
|
|
|
|
|
fi
|
2020-02-17 15:53:54 +08:00
|
|
|
exit 0
|
