Commit fd4a35a8 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

extract_idmap xvector on CPU

parent 24d726cf
......@@ -642,7 +642,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