Commit 9ff94c59 authored by Anthony Larcher's avatar Anthony Larcher
Browse files


parent 1873385c
......@@ -1067,7 +1067,7 @@ def get_loaders(dataset_opts, training_opts, model_opts, speaker_number):
Set the dataloaders according to the dataset_yaml
First we load the dataframe from CSV file in order to split it for training and validation purpose
Then we provide those two
Then we provide those two
df = pandas.read_csv(dataset_opts["dataset_csv"])
......@@ -1294,7 +1294,7 @@ def new_xtrain(dataset_description,
# Display the entire configurations as YAML dictionnaries
# Display the entire configurations as YAML dictionaries
if local_rank < 1:"\n*********************************\nDataset options\n*********************************\n"), default_flow_style=False))
......@@ -1309,11 +1309,12 @@ def new_xtrain(dataset_description,
embedding_size = model.embedding_size
# Set the device and manage parallel processing
device = torch.cuda.device(local_rank)
#device = torch.cuda.device(local_rank)
device = torch.device(local_rank)
# If multi-gpu
""" [HOW TO] from
- Add the following line right after "if __name__ == '__main__':" in your main script :
parser.add_argument('--local_rank', type=int, default=-1, metavar='N', help='Local process rank.')
Supports Markdown
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