mirror of
https://github.com/facebookresearch/fairseq.git
synced 2024-10-26 17:32:57 +03:00
Update raw_audio_dataset.py
This commit is contained in:
parent
e853a2d157
commit
4eac4a0155
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user