DeepLearningExamples/PyTorch/SpeechSynthesis/Tacotron2/scripts/train_tacotron2.sh

3 lines
258 B
Bash

mkdir -p output
python -m multiproc train.py -m Tacotron2 -o ./output/ -lr 1e-3 --epochs 1500 -bs 80 --weight-decay 1e-6 --grad-clip-thresh 1.0 --cudnn-benchmark=True --log-file ./output/nvlog.json --anneal-steps 500 1000 1500 --anneal-factor 0.1 --fp16-run