fix speaker encoder model initialization

2026-09-09 11:16:00 +08:00 · 2021-05-11 11:29:18 +02:00
parent 10de40bba1
commit 18e76a2309
1 changed files with 2 additions and 2 deletions
@@ -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