Merge pull request #154 from miccio-dk/master

specify device for RandomCrop output variable
asteroid-team · Aug 3, 2022 · 1590056 · 1590056
2 parents 1e74ce8 + 0c3b0b9
commit 1590056
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch_audiomentations/augmentations/random_crop.py b/torch_audiomentations/augmentations/random_crop.py
@@ -71,7 +71,8 @@ def forward(self, samples, sampling_rate: typing.Optional[int] = None):
             0, samples.shape[2] - self.num_samples, (samples.shape[2],)
         )
         samples_cropped = torch.empty(
-            (samples.shape[0], samples.shape[1], self.num_samples)
+            (samples.shape[0], samples.shape[1], self.num_samples),
+            device=samples.device
         )
         for i, sample in enumerate(samples):