facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-10-26 17:32:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update raw_audio_dataset.py

Browse Source
This commit is contained in:
Hanbyul Kim 2023-08-01 17:57:22 +09:00 committed by GitHub
parent e853a2d157
commit 4eac4a0155
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fairseq/data/audio/raw_audio_dataset.py
View File

@ -175,7 +175,7 @@ class RawAudioDataset(FairseqDataset):
return out
def _get_mask_indices_dims(self, size, padding=0, dilation=1):
if size not in self.feature_encoder_spec:
if size not in self._features_size_map:
L_in = size
for (_, kernel_size, stride) in self.feature_encoder_spec:
L_out = L_in + 2 * padding - dilation * (kernel_size - 1) - 1