long xvector segment limit

nnet/xvector.py

@@ -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()