Commit 383393c6 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

cleaning

parent b17d1364
...@@ -960,6 +960,7 @@ def get_network(model_opts, local_rank): ...@@ -960,6 +960,7 @@ def get_network(model_opts, local_rank):
""" """
:param model_opts: :param model_opts:
:param local_rank:
:return: :return:
""" """
...@@ -1011,7 +1012,9 @@ def get_network(model_opts, local_rank): ...@@ -1011,7 +1012,9 @@ def get_network(model_opts, local_rank):
def get_loaders(dataset_opts, training_opts, model_opts): def get_loaders(dataset_opts, training_opts, model_opts):
""" """
:param dataset_yaml: :param dataset_opts:
:param training_opts:
:param model_opts:
:return: :return:
""" """
...@@ -1259,8 +1262,7 @@ def xtrain(dataset_description, ...@@ -1259,8 +1262,7 @@ def xtrain(dataset_description,
monitor.logger.info(yaml.dump(training_opts, default_flow_style=False)) monitor.logger.info(yaml.dump(training_opts, default_flow_style=False))
# Initialize the model # Initialize the model
model = get_network(model_opts) model = get_network(model_opts, local_rank)
#speaker_number = model.speaker_number
embedding_size = model.embedding_size embedding_size = model.embedding_size
# Set the device and manage parallel processing # Set the device and manage parallel processing
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment