16 lines
321 B
Bash
16 lines
321 B
Bash
|
|
# train model
|
|
export CUDA_VISIBLE_DEVICES=0
|
|
python -u train_transformer.py \
|
|
--use_gpu=1 \
|
|
--data='../../dataset/LJSpeech-1.1' \
|
|
--output='./experiment' \
|
|
--config='configs/ljspeech.yaml' \
|
|
#--checkpoint='./checkpoint/transformer/step-120000' \
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Failed in training!"
|
|
exit 1
|
|
fi
|
|
exit 0
|