LydiaSanyu
commited on
Commit
·
21bef79
1
Parent(s):
10564d2
Update README.md
Browse files
README.md
CHANGED
@@ -67,23 +67,4 @@ mel_outputs, mel_lengths, alignments = tacotron2.encode_batch(items)
|
|
67 |
```
|
68 |
|
69 |
### Inference on GPU
|
70 |
-
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
71 |
-
|
72 |
-
### Training
|
73 |
-
The model was trained with SpeechBrain.
|
74 |
-
To train it from scratch follow these steps:
|
75 |
-
1. Clone SpeechBrain:
|
76 |
-
```bash
|
77 |
-
git clone https://github.com/speechbrain/speechbrain/
|
78 |
-
```
|
79 |
-
2. Install it:
|
80 |
-
```bash
|
81 |
-
cd speechbrain
|
82 |
-
pip install -r requirements.txt
|
83 |
-
pip install -e .
|
84 |
-
```
|
85 |
-
3. Run Training:
|
86 |
-
```bash
|
87 |
-
cd recipes/LJSpeech/TTS/tacotron2/
|
88 |
-
python train.py --device=cuda:0 --max_grad_norm=1.0 --data_folder=/your_folder/LJSpeech-1.1 hparams/train.yaml
|
89 |
-
```
|
|
|
67 |
```
|
68 |
|
69 |
### Inference on GPU
|
70 |
+
To perform inference on the GPU, add `run_opts={"device":"cuda"}` when calling the `from_hparams` method.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|