add speaker encoder parameters in config.json, for more easy changes

2026-09-10 11:50:20 +08:00 · 2020-08-05 19:33:35 +02:00
parent cb212b8049
commit bd4c6ee42a
1 changed files with 4 additions and 4 deletions
@@ -135,10 +135,10 @@ def main(args):  # pylint: disable=redefined-outer-name
    global meta_data_eval

    ap = AudioProcessor(**c.audio)
-    model = SpeakerEncoder(input_dim=40,
-                           proj_dim=128,
-                           lstm_dim=384,
-                           num_lstm_layers=3)
+    model = SpeakerEncoder(input_dim=c.model['input_dim'],
+                           proj_dim=c.model['proj_dim'],
+                           lstm_dim=c.model['lstm_dim'],
+                           num_lstm_layers=c.model['num_lstm_layers'])
    optimizer = RAdam(model.parameters(), lr=c.lr)

    if c.loss == "ge2e":