Browse Source

README.md: adding explanation on training from pre-trained model

master
rafaelvalle 5 years ago
parent
commit
bff304f432
1 changed files with 8 additions and 0 deletions
  1. +8
    -0
      README.md

+ 8
- 0
README.md View File

@ -33,6 +33,13 @@ Visit our [website] for audio samples using our published [Tacotron 2] and
1. `python train.py --output_directory=outdir --log_directory=logdir` 1. `python train.py --output_directory=outdir --log_directory=logdir`
2. (OPTIONAL) `tensorboard --logdir=outdir/logdir` 2. (OPTIONAL) `tensorboard --logdir=outdir/logdir`
## Training using a pre-trained model
Training using a pre-trained model can lead to faster convergence
By default, the dataset dependent text embedding layers are [ignored]
1. Download our published [Tacotron 2] model
2. `python train.py --output_directory=outdir --log_directory=logdir -c tacotron2_statedict.pt --warm_start`
## Multi-GPU (distributed) and FP16 Training ## Multi-GPU (distributed) and FP16 Training
1. `python -m multiproc train.py --output_directory=outdir --log_directory=logdir --hparams=distributed_run=True,fp16_run=True` 1. `python -m multiproc train.py --output_directory=outdir --log_directory=logdir --hparams=distributed_run=True,fp16_run=True`
@ -69,3 +76,4 @@ Wang and Zongheng Yang.
[Tacotron 2]: https://drive.google.com/file/d/1c5ZTuT7J08wLUoVZ2KkUs_VdZuJ86ZqA/view?usp=sharing [Tacotron 2]: https://drive.google.com/file/d/1c5ZTuT7J08wLUoVZ2KkUs_VdZuJ86ZqA/view?usp=sharing
[pytorch 1.0]: https://github.com/pytorch/pytorch#installation [pytorch 1.0]: https://github.com/pytorch/pytorch#installation
[website]: https://nv-adlr.github.io/WaveGlow [website]: https://nv-adlr.github.io/WaveGlow
[ignored]: https://github.com/NVIDIA/tacotron2/blob/master/hparams.py#L22

Loading…
Cancel
Save