ShiromiyaGamer
commited on
Commit
•
934fe4e
1
Parent(s):
b66714e
Upload dump_w2v2_feature.py
Browse files- dump_w2v2_feature.py +2 -1
dump_w2v2_feature.py
CHANGED
@@ -65,7 +65,8 @@ class Wav2Vec2FeatureReader(object):
|
|
65 |
padding_mask=None,
|
66 |
mask=False,
|
67 |
)
|
68 |
-
|
|
|
69 |
feat.append(feat_chunk)
|
70 |
return torch.cat(feat, 1).squeeze(0)
|
71 |
|
|
|
65 |
padding_mask=None,
|
66 |
mask=False,
|
67 |
)
|
68 |
+
# Supondo que o primeiro elemento da tupla seja o que você precisa
|
69 |
+
feat_chunk = res[0] # Ajuste o índice conforme necessário
|
70 |
feat.append(feat_chunk)
|
71 |
return torch.cat(feat, 1).squeeze(0)
|
72 |
|