Skip to content

Commit 8cbac00

Browse files
Merge pull request #719 from NVIDIA/gh/release
[FastPitch/PyT] Adding notebooks
2 parents d1b1854 + b2c72b2 commit 8cbac00

File tree

11 files changed

+899
-24
lines changed

11 files changed

+899
-24
lines changed

PyTorch/SpeechSynthesis/FastPitch/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:20.08-py3
1+
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:20.09-py3
22
FROM ${FROM_IMAGE_NAME}
33

44
ADD requirements.txt .

PyTorch/SpeechSynthesis/FastPitch/README.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ The following section lists the requirements that you need to meet in order to s
229229

230230
This repository contains Dockerfile which extends the PyTorch NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following components:
231231
- [NVIDIA Docker](https://github.com/NVIDIA/nvidia-docker)
232-
- [PyTorch 20.06-py3 NGC container](https://ngc.nvidia.com/registry/nvidia-pytorch)
232+
- [PyTorch 20.09-py3 NGC container](https://ngc.nvidia.com/registry/nvidia-pytorch)
233233
or newer
234234
- supported GPUs:
235235
- [NVIDIA Volta architecture](https://www.nvidia.com/en-us/data-center/volta-gpu-architecture/)
@@ -246,7 +246,7 @@ For those unable to use the PyTorch NGC container, to set up the required enviro
246246

247247
## Quick Start Guide
248248

249-
To train your model using mixed or TF32 precision with Tensor Cores or using FP32, perform the following steps using the default parameters of the FastPitch model on the LJSpeech 1.1 dataset. For the specifics concerning training and inference, see the [Advanced](#advanced) section.
249+
To train your model using mixed or TF32 precision with Tensor Cores or using FP32, perform the following steps using the default parameters of the FastPitch model on the LJSpeech 1.1 dataset. For the specifics concerning training and inference, see the [Advanced](#advanced) section. Pre-trained FastPitch models are available for download on [NGC](https://ngc.nvidia.com/catalog/models?query=FastPitch&quickFilter=models).
250250

251251
1. Clone the repository.
252252
```bash
@@ -502,6 +502,8 @@ text as input and runs FastPitch and then WaveGlow inference to produce an
502502
audio file. It requires pre-trained checkpoints of both models
503503
and input text as a text file, with one phrase per line.
504504
505+
Pre-trained FastPitch models are available for download on [NGC](https://ngc.nvidia.com/catalog/models?query=FastPitch&quickFilter=models).
506+
505507
Having pre-trained models in place, run the sample inference on LJSpeech-1.1 test-set with:
506508
```bash
507509
bash scripts/inference_example.sh

0 commit comments

Comments
 (0)