Commit 2a32695a authored by Anthony Larcher's avatar Anthony Larcher
Browse files

parallel

parent 9ff94c59
...@@ -1056,7 +1056,7 @@ def get_network(model_opts): ...@@ -1056,7 +1056,7 @@ def get_network(model_opts):
return model return model
def get_loaders(dataset_opts, training_opts, model_opts, speaker_number): def get_loaders(dataset_opts, training_opts, model_opts):
""" """
:param dataset_yaml: :param dataset_yaml:
...@@ -1100,14 +1100,14 @@ def get_loaders(dataset_opts, training_opts, model_opts, speaker_number): ...@@ -1100,14 +1100,14 @@ def get_loaders(dataset_opts, training_opts, model_opts, speaker_number):
batch_size = dataset_opts["batch_size"]//torch.cuda.device_count() batch_size = dataset_opts["batch_size"]//torch.cuda.device_count()
side_sampler = SideSampler(training_set.sessions['speaker_idx'], side_sampler = SideSampler(training_set.sessions['speaker_idx'],
speaker_number, model_opts["speaker_number"],
dataset_opts["train"]["sampler"]["examples_per_speaker"], dataset_opts["train"]["sampler"]["examples_per_speaker"],
dataset_opts["train"]["sampler"]["samples_per_speaker"], dataset_opts["train"]["sampler"]["samples_per_speaker"],
dataset_opts["batch_size"]) dataset_opts["batch_size"])
else: else:
batch_size = dataset_opts["batch_size"] batch_size = dataset_opts["batch_size"]
side_sampler = SideSampler(training_set.sessions['speaker_idx'], side_sampler = SideSampler(training_set.sessions['speaker_idx'],
speaker_number, model_opts["speaker_number"],
samples_per_speaker, samples_per_speaker,
batch_size, batch_size,
batch_size, batch_size,
......
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