Skip to content

Commit

Permalink
[src] Update overlap_add.py, auto casting device for window (#675)
Browse files Browse the repository at this point in the history
allows to use LambdaOverlapAdd also on GPU without calling before .to on the instantiated module.
  • Loading branch information
popcornell committed Sep 3, 2023
1 parent cd5b35d commit 68c2669
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion asteroid/dsp/overlap_add.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 68c2669

Please sign in to comment.