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):
"""
: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
......
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