maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

adding support for --cpu-run

Browse source
This commit is contained in:
maggiezha 2020-05-06 21:48:43 +10:00 committed by GitHub
parent 0e986cc1f0
commit 342c4710fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
PyTorch/SpeechSynthesis/Tacotron2/README.md
View file

@ -323,6 +323,7 @@ inference using the respective checkpoints that are passed as `--tacotron2`
and `--waveglow` arguments. Tacotron2 and WaveGlow checkpoints can also be downloaded from NGC:
https://ngc.nvidia.com/catalog/models/nvidia:tacotron2pyt_fp16/files?version=3
https://ngc.nvidia.com/catalog/models/nvidia:waveglow256pyt_fp16/files?version=2
To run inference issue:
@ -338,7 +339,7 @@ and `--waveglow` arguments. Tacotron2 and WaveGlow checkpoints can also be downl
You can also run inference on CPU with TorchScript by adding flag --cpu-run:
```
```bash
export CUDA_VISIBLE_DEVICES=
```
```bash