mwirth7 commited on
Commit
38e413a
·
verified ·
1 Parent(s): 3e33e4b

Update feature_extractor.py

Browse files
Files changed (1) hide show
  1. feature_extractor.py +1 -1
feature_extractor.py CHANGED
@@ -66,7 +66,7 @@ class BirdMAEFeatureExtractor(SequenceFeatureExtractor):
66
  ):
67
 
68
  if not torch.is_tensor(waveform_batch):
69
- waveform_batch = torch.tensor(waveform_batch)
70
 
71
  if len(waveform_batch.shape) == 1:
72
  waveform_batch = waveform_batch.unsqueeze(0)
 
66
  ):
67
 
68
  if not torch.is_tensor(waveform_batch):
69
+ waveform_batch = torch.from_numpy(np.array(waveform_batch))
70
 
71
  if len(waveform_batch.shape) == 1:
72
  waveform_batch = waveform_batch.unsqueeze(0)