fix speaker encoder model initialization

2021-05-10 15:11:28 +02:00 · 2021-05-10 15:11:28 +02:00 · 18e76a2309
parent 10de40bba1
commit 18e76a2309
1 changed files with 2 additions and 2 deletions
--- a/TTS/speaker_encoder/utils/generic_utils.py
+++ b/TTS/speaker_encoder/utils/generic_utils.py
@ -9,6 +9,6 @@ def to_camel(text):


 def setup_model(c):
-    model = SpeakerEncoder(c.model["input_dim"], c.model["proj_dim"],
-                           c.model["lstm_dim"], c.model["num_lstm_layers"])
+    model = SpeakerEncoder(c.model_params["input_dim"], c.model_params["proj_dim"],
+                           c.model_params["lstm_dim"], c.model_params["num_lstm_layers"])
    return model