Commit 41957818 authored by Anthony Larcher's avatar Anthony Larcher
Browse files
parents 5b4f0582 05c02acb
......@@ -447,7 +447,8 @@ def extract_idmap(args, device_ID, segment_indices, fs_params, idmap_name, outpu
Function that takes a model and an idmap and extract all x-vectors based on this model
and return a StatServer containing the x-vectors
"""
device = torch.device("cuda:{}".format(device_ID))
#device = torch.device("cuda:{}".format(device_ID))
device = torch.device('cpu')
# Create the dataset
tmp_idmap = IdMap(idmap_name)
......
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