diff --git a/README.md b/README.md index 53ec1aada..d2bfbeee6 100644 --- a/README.md +++ b/README.md @@ -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. diff --git a/TTS/tts/configs/config.json b/TTS/tts/configs/config.json index be6e689e8..1b63b0374 100644 --- a/TTS/tts/configs/config.json +++ b/TTS/tts/configs/config.json @@ -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