bug fix issue #20

612ef0b3 · Maxime Burchi · b28a0aaa · 612ef0b3
Commit 612ef0b3 authored Feb 10, 2023 by Maxime Burchi
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

models/modules.py models/modules.py +1 -1

No files found.
--- a/models/modules.py
+++ b/models/modules.py
@@ -146,7 +146,7 @@ class SpecAugment(nn.Module):
            for b in range(x.size(0)):
                T = int(self.pS * x_len[b])
                for _ in range(self.mT):
-                    x[b, :, :x_len[b]] = torchaudio.transforms.TimeMasking(time_mask_param=T).forward(x[b, :, :x_len[b]])
+                    x[b:b+1, :, :x_len[b]] = torchaudio.transforms.TimeMasking(time_mask_param=T).forward(x[b:b+1, :, :x_len[b]])

        return x