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

minor bug

parent 176f4337
......@@ -1803,7 +1803,7 @@ def extract_embeddings_per_speaker(idmap_name,
for idx, (data, mod, seg, start, stop) in enumerate(tqdm.tqdm(dataloader, desc='xvector extraction', mininterval=1)):
if data.shape[1] > 20000000:
data = data[..., :20000000]
vec = model(data.to(device), is_eval=True)
_, vec = model(x=data.to(device), is_eval=True, norm_embedding=True)
embeddings.stat1[idx, :] = vec.detach().cpu()
return embeddings
......
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