handle facebookresearch#21

seungwonpark · Nov 18, 2021 · 6601da3 · 6601da3
1 parent 1417166
commit 6601da3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpc/dataset.py b/cpc/dataset.py
@@ -410,8 +410,8 @@ def __iter__(self):
 
 def extractLength(couple):
     speaker, locPath = couple
-    info = torchaudio.info(str(locPath))[0]
-    return info.length
+    info = torchaudio.info(str(locPath))
+    return info.num_frames
 
 
 def findAllSeqs(dirName,