Use pytorch randint for ResizeSignalpt (#103)

See this comment for the issue: #96 (comment)
fastaudio · Sep 29, 2021 · 3d6c0a0 · 3d6c0a0
1 parent e7a9caa
commit 3d6c0a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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