Upload dump_w2v2_feature.py
Browse files- dump_w2v2_feature.py +1 -1
dump_w2v2_feature.py
CHANGED
|
@@ -66,7 +66,7 @@ class Wav2Vec2FeatureReader(object):
|
|
| 66 |
mask=False,
|
| 67 |
)
|
| 68 |
# Supondo que o primeiro elemento da tupla seja o que você precisa
|
| 69 |
-
feat_chunk = res[
|
| 70 |
feat.append(feat_chunk)
|
| 71 |
return torch.cat(feat, 1).squeeze(0)
|
| 72 |
|
|
|
|
| 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 |
|