Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add `gradual_training` argument to tacotron.py

This commit is contained in:
Eren Gölge 2021-05-25 10:42:16 +02:00
parent 5a2e75f0ee
commit bdbfc95618
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/tts/models/tacotron.py
View File

@ -47,8 +47,9 @@ class Tacotron(TacotronAbstract):
gst (Coqpit, optional): Coqpit to initialize the GST module. If `None`, GST is disabled. Defaults to None. gst (Coqpit, optional): Coqpit to initialize the GST module. If `None`, GST is disabled. Defaults to None.
memory_size (int, optional): size of the history queue fed to the prenet. Model feeds the last ```memory_size``` memory_size (int, optional): size of the history queue fed to the prenet. Model feeds the last ```memory_size```
output frames to the prenet. output frames to the prenet.
gradual_trainin (List): Gradual training schedule. If None or `[]`, no gradual training is used.
Defaults to `[]`.
""" """
def __init__( def __init__(
self, self,
num_chars, num_chars,
@ -77,6 +78,7 @@ class Tacotron(TacotronAbstract):
use_gst=False, use_gst=False,
gst=None, gst=None,
memory_size=5, memory_size=5,
gradual_training=[]
): ):
super().__init__( super().__init__(
num_chars, num_chars,
@ -104,6 +106,7 @@ class Tacotron(TacotronAbstract):
speaker_embedding_dim, speaker_embedding_dim,
use_gst, use_gst,
gst, gst,
gradual_training
) )
# speaker embedding layers # speaker embedding layers