clean

nnet/xvector.py

@@ -1023,6 +1023,10 @@ def get_network(model_opts, local_rank):
             if name.split(".")[0] in model_opts["reset_parts"]:
                 param.requires_grad = False
 
+    if model_opts["loss"]["type"] == "aam" and not (model_opts["loss"]["aam_margin"] == 0.2 and model_opts["loss"]["aam_s"] == 30):
+        model.after_speaker_embedding.change_params(model_opts["loss"]["aam_s"], model_opts["loss"]["aam_margin"])
+        print(f"Modified AAM: margin = {model.after_speaker_embedding.m} and s = {model.after_speaker_embedding.s}")
+
     if local_rank < 1:
         logging.info(model)
         logging.info("Model_parameters_count: {:d}".format(