Skip to content

Commit

Permalink
Merge pull request #546 from ayush-1506/patch-1
Browse files Browse the repository at this point in the history
Fix readme and config file
  • Loading branch information
Eren Gölge committed Oct 21, 2020
2 parents 98f15e1 + 2a3559f commit af6f862
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
6 changes: 3 additions & 3 deletions README.md
Expand Up @@ -151,15 +151,15 @@ tail -n 1100 metadata_shuf.csv > metadata_val.csv

To train a new model, you need to define your own ```config.json``` file (check the example) and call with the command below. You also set the model architecture in ```config.json```.

```python TTS/bin/train.py --config_path TTS/tts/configs/config.json```
```python TTS/bin/train_tts.py --config_path TTS/tts/configs/config.json```

To fine-tune a model, use ```--restore_path```.

```python TTS/bin/train.py --config_path TTS/tts/configs/config.json --restore_path /path/to/your/model.pth.tar```
```python TTS/bin/train_tts.py --config_path TTS/tts/configs/config.json --restore_path /path/to/your/model.pth.tar```

To continue an old training run, use ```--continue_path```.

```python TTS/bin/train.py --continue_path /path/to/your/run_folder/```
```python TTS/bin/train_tts.py --continue_path /path/to/your/run_folder/```

For multi-GPU training use ```distribute.py```. It enables process based multi-GPU training where each process uses a single GPU.

Expand Down
3 changes: 2 additions & 1 deletion TTS/tts/configs/config.json
Expand Up @@ -147,7 +147,8 @@
// with the dictionary being len(dict) <= len(gst_style_tokens).
"gst_embedding_dim": 512,
"gst_num_heads": 4,
"gst_style_tokens": 10
"gst_style_tokens": 10,
"gst_use_speaker_embedding": false
},

// DATASETS
Expand Down

0 comments on commit af6f862

Please sign in to comment.