2020-02-13 14:53:45 +08:00
|
|
|
|
|
|
|
|
# train model
|
2020-04-09 20:06:04 +08:00
|
|
|
export CUDA_VISIBLE_DEVICES=0
|
2020-02-13 14:53:45 +08:00
|
|
|
python -u train_transformer.py \
|
|
|
|
|
--use_gpu=1 \
|
2020-04-09 20:06:04 +08:00
|
|
|
--data='../../dataset/LJSpeech-1.1' \
|
|
|
|
|
--output='./experiment' \
|
|
|
|
|
--config='configs/ljspeech.yaml' \
|
|
|
|
|
#--checkpoint='./checkpoint/transformer/step-120000' \
|
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
|
