From bdbfc95618dc5ba1bb10dc29bf1efdd510b8f9e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 25 May 2021 10:42:16 +0200
Subject: [PATCH] add `gradual_training` argument to tacotron.py

---
 TTS/tts/models/tacotron.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/TTS/tts/models/tacotron.py b/TTS/tts/models/tacotron.py
index 4413b015..c1d95a25 100644
--- a/TTS/tts/models/tacotron.py
+++ b/TTS/tts/models/tacotron.py
@@ -47,8 +47,9 @@ class Tacotron(TacotronAbstract):
         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```
             output frames to the prenet.
+        gradual_trainin (List): Gradual training schedule. If None or `[]`, no gradual training is used.
+            Defaults to `[]`.
     """
-
     def __init__(
         self,
         num_chars,
@@ -77,6 +78,7 @@ class Tacotron(TacotronAbstract):
         use_gst=False,
         gst=None,
         memory_size=5,
+        gradual_training=[]
     ):
         super().__init__(
             num_chars,
@@ -104,6 +106,7 @@ class Tacotron(TacotronAbstract):
             speaker_embedding_dim,
             use_gst,
             gst,
+            gradual_training
         )
 
         # speaker embedding layers