ShiromiyaGamer commited on
Commit
3ef1427
1 Parent(s): 934fe4e

Upload dump_w2v2_feature.py

Browse files
Files changed (1) hide show
  1. 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[0] # Ajuste o índice conforme necessário
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[x] # Ajuste o índice conforme necessário
70
  feat.append(feat_chunk)
71
  return torch.cat(feat, 1).squeeze(0)
72