From 68c26692da9bfc545e8cae0a9a650296dce34c60 Mon Sep 17 00:00:00 2001 From: Samuele Cornell Date: Sun, 3 Sep 2023 13:33:45 +0100 Subject: [PATCH] [src] Update overlap_add.py, auto casting device for window (#675) allows to use LambdaOverlapAdd also on GPU without calling before .to on the instantiated module. --- asteroid/dsp/overlap_add.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/asteroid/dsp/overlap_add.py b/asteroid/dsp/overlap_add.py index 256b43bc5..c858e6b4e 100644 --- a/asteroid/dsp/overlap_add.py +++ b/asteroid/dsp/overlap_add.py @@ -113,7 +113,7 @@ def ola_forward(self, x): frame = _reorder_sources(frame, out[-1], n_src, self.window_size, self.hop_size) if self.use_window: - frame = frame * self.window + frame = frame * self.window.to(frame) else: frame = frame / (self.window_size / self.hop_size) out.append(frame)