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, ...@@ -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)): for idx, (data, mod, seg, start, stop) in enumerate(tqdm.tqdm(dataloader, desc='xvector extraction', mininterval=1)):
if data.shape[1] > 20000000: if data.shape[1] > 20000000:
data = data[..., :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() embeddings.stat1[idx, :] = vec.detach().cpu()
return embeddings 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