Commit 8ea51326 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

long xvector segment limit

parent 47e78263
......@@ -1082,6 +1082,8 @@ def extract_embeddings(idmap_name,
# Process the data
with torch.no_grad():
for idx, (data, mod, seg, start, stop) in tqdm.tqdm(enumerate(dataloader)):
if data.shape[1] > 20000000:
data = data[...,:20000000]
vec = model(data.to(device), is_eval=True)
embeddings.stat1[idx, :] = vec.detach().cpu()
......
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