cleaning

nnet/xvector.py

@@ -960,6 +960,7 @@ def get_network(model_opts, local_rank):
     """
 
     :param model_opts:
+    :param local_rank:
     :return:
     """
 
@@ -1011,7 +1012,9 @@ def get_network(model_opts, local_rank):
 def get_loaders(dataset_opts, training_opts, model_opts):
     """
 
-    :param dataset_yaml:
+    :param dataset_opts:
+    :param training_opts:
+    :param model_opts:
     :return:
     """
 
@@ -1259,8 +1262,7 @@ def xtrain(dataset_description,
         monitor.logger.info(yaml.dump(training_opts, default_flow_style=False))
 
     # Initialize the model
-    model = get_network(model_opts)
-    #speaker_number = model.speaker_number
+    model = get_network(model_opts, local_rank)
     embedding_size = model.embedding_size
 
     # Set the device and manage parallel processing