Cast tensor in Delay to audio tensor's device

Spijkervet · web-flow · commit f3e0ad837acb · 2021-10-18T16:18:45.000+02:00
diff --git a/torchaudio_augmentations/augmentations/delay.py b/torchaudio_augmentations/augmentations/delay.py
@@ -28,7 +28,7 @@ def forward(self, audio):
         )
 
         offset = self.calc_offset(ms)
-        beginning = torch.zeros(audio.shape[0], offset)
+        beginning = torch.zeros(audio.shape[0], offset).to(audio.device)
         end = audio[:, :-offset]
         delayed_signal = torch.cat((beginning, end), dim=1)
         delayed_signal = delayed_signal * self.volume_factor

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def forward(self, audio):`
`28`	`28`	`)`
`29`	`29`
`30`	`30`	`offset = self.calc_offset(ms)`
`31`		`- beginning = torch.zeros(audio.shape[0], offset)`
	`31`	`+ beginning = torch.zeros(audio.shape[0], offset).to(audio.device)`
`32`	`32`	`end = audio[:, :-offset]`
`33`	`33`	`delayed_signal = torch.cat((beginning, end), dim=1)`
`34`	`34`	`delayed_signal = delayed_signal * self.volume_factor`