fix the speaker_encoder import

wespeaker/models/redimnet.py

@@ -1008,14 +1008,14 @@ def ReDimNetB6(feat_dim=72, embed_dim=192, pooling_func="ASTP", two_emb_layer=Fa
 
 
 if __name__ == "__main__":
-    # x = torch.zeros(1, 200, 72)
-    # model = ReDimNet(feat_dim=72, embed_dim=192, two_emb_layer=False)
-    # model.eval()
-    # out = model(x)
-    # print(out[-1].size())
-
-    # num_params = sum(p.numel() for p in model.parameters())
-    # print("{} M".format(num_params / 1e6))
+    x = torch.zeros(1, 200, 72)
+    model = ReDimNet(feat_dim=72, embed_dim=192, two_emb_layer=False)
+    model.eval()
+    out = model(x)
+    print(out[-1].size())
+
+    num_params = sum(p.numel() for p in model.parameters())
+    print("{} M".format(num_params / 1e6))
 
     # Currently, the model sizes differ from the ones in the paper
     model_classes = [

wespeaker/models/speaker_model.py

@@ -20,6 +20,7 @@
 import wespeaker.models.eres2net as eres2net
 import wespeaker.models.gemini_dfresnet as gemini
 import wespeaker.models.res2net as res2net
+import wespeaker.models.redimnet as redimnet
 
 
 def get_speaker_model(model_name: str):
@@ -39,6 +40,8 @@ def get_speaker_model(model_name: str):
         return getattr(res2net, model_name)
     elif model_name.startswith("Gemini"):
         return getattr(gemini, model_name)
+    elif model_name.startswith("ReDimNet"):
+        return getattr(redimnet, model_name)
     else:  # model_name error !!!
         print(model_name + " not found !!!")
         exit(1)