From 3d6c0a0cb2f6eb5868fcb22a02882b06c8372a6f Mon Sep 17 00:00:00 2001 From: Filip Mulier <3823334+filipmu@users.noreply.github.com> Date: Wed, 29 Sep 2021 10:44:38 -0500 Subject: [PATCH] Use pytorch randint for ResizeSignalpt (#103) See this comment for the issue: https://github.com/fastaudio/fastaudio/issues/96#issuecomment-846467049 --- src/fastaudio/augment/signal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/fastaudio/augment/signal.py b/src/fastaudio/augment/signal.py index ac25eed..0eeb1e2 100644 --- a/src/fastaudio/augment/signal.py +++ b/src/fastaudio/augment/signal.py @@ -44,7 +44,7 @@ def encodes(self, ai: AudioTensor) -> AudioTensor: elif orig_samples < crop_samples: ai.data = _tfm_pad_signal(sig, crop_samples, pad_mode=self.pad_mode) else: - crop_start = random.randint(0, int(orig_samples - crop_samples)) + crop_start = torch.randint(0, int(orig_samples - crop_samples),(1,)).item() ai.data = sig[:, crop_start : crop_start + crop_samples] return ai