Fixed bug in MaskedSpecAug that overestimates samples. (#6775)

Signed-off-by: tbartley94 <[email protected]>
NVIDIA · Jun 6, 2023 · 010a0e6 · 010a0e6
1 parent 3063e32
commit 010a0e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/modules/audio_preprocessing.py b/nemo/collections/asr/modules/audio_preprocessing.py
@@ -608,7 +608,7 @@ def forward(self, input_spec, length):
 
         for idx in range(input_spec.shape[0]):
             cur_len = length[idx]
-            patches = range(cur_len // self.patch_size - 1)
+            patches = range(cur_len // self.patch_size)
             masked_patches = random.sample(patches, mask_patches)
 
             for mp in masked_patches: