compute and add style tokens in gst

This commit is contained in:
Eren Golge 2019-09-21 09:58:58 +02:00
parent 6561013d28
commit 1f4ec804b6
2 changed files with 2 additions and 1 deletions

View File

@ -13,4 +13,5 @@ python3 setup.py develop
# python3 distribute.py --config_path config.json --data_path /data/ro/shared/data/keithito/LJSpeech-1.1/
# cp -R ${USER_DIR}/Mozilla_22050 ../tmp/
# python3 distribute.py --config_path config_tacotron_gst.json --data_path ../tmp/Mozilla_22050/
python3 distribute.py --config_path config.json --data_path /data/rw/home/LibriTTS/train-clean-360
while true; do sleep 1000000; done

View File

@ -40,7 +40,7 @@ class TacotronGST(Tacotron):
location_attn,
separate_stopnet)
gst_embedding_dim = 256
decoder_dim = 512 + gst_embedding_dim if num_speakers > 1 else 256 + gst_embedding_dim
decoder_dim = 512 if num_speakers > 1 else 256
proj_speaker_dim = 80 if num_speakers > 1 else 0
self.decoder = Decoder(decoder_dim, mel_dim, r, memory_size, attn_win,
attn_norm, prenet_type, prenet_dropout,