type def for gradual_training

2021-05-19 14:00:44 +02:00 · 2021-05-19 14:00:44 +02:00 · df6a98d0c3
parent 16576d6408
commit df6a98d0c3
1 changed files with 1 additions and 1 deletions
--- a/TTS/tts/configs/tacotron_config.py
+++ b/TTS/tts/configs/tacotron_config.py
@ -122,7 +122,7 @@ class TacotronConfig(BaseTTSConfig):
    gst_style_input: str = None
    # model specific params
    r: int = 2
-    gradual_training: List[List] = None
+    gradual_training: List[List[int]] = None
    memory_size: int = -1
    prenet_type: str = "original"
    prenet_dropout: bool = True