fix add colored noise to prevent issue.

asteroid-team · Aug 29, 2024 · 9cb06d4 · 9cb06d4
1 parent a5c584d
commit 9cb06d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_audiomentations/augmentations/colored_noise.py b/torch_audiomentations/augmentations/colored_noise.py
@@ -27,7 +27,7 @@ def _gen_noise(f_decay, num_samples, sample_rate, device):
     )
     spec *= mask
     noise = Audio.rms_normalize(irfft(spec).unsqueeze(0)).squeeze()
-    noise = torch.cat([noise] * int(ceil(num_samples / sample_rate)))
+    noise = torch.cat([noise] * int(ceil(num_samples / noise.shape[0])))
     return noise[:num_samples]