From 22c8e9c9112e0475eb18b71fc2c62ff16ab04f85 Mon Sep 17 00:00:00 2001 From: ksanjeevan Date: Thu, 5 Mar 2020 10:39:38 -0800 Subject: [PATCH 1/2] phase_advance should be a buffer so it moves device correctly --- torchaudio/transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchaudio/transforms.py b/torchaudio/transforms.py index d807b02365..ecc9c54506 100644 --- a/torchaudio/transforms.py +++ b/torchaudio/transforms.py @@ -610,8 +610,8 @@ def __init__(self, hop_length=None, n_freq=201, fixed_rate=None): n_fft = (n_freq - 1) * 2 hop_length = hop_length if hop_length is not None else n_fft // 2 - self.phase_advance = torch.linspace(0, math.pi * hop_length, n_freq)[..., None] - + self.register_buffer('phase_advance', torch.linspace(0, math.pi * hop_length, n_freq)[..., None]) + def forward(self, complex_specgrams, overriding_rate=None): # type: (Tensor, Optional[float]) -> Tensor r""" From a8557328b585bd825ff8a3fb23e4adca585c042e Mon Sep 17 00:00:00 2001 From: Vincent QB Date: Thu, 5 Mar 2020 13:54:38 -0500 Subject: [PATCH 2/2] flake8 --- torchaudio/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchaudio/transforms.py b/torchaudio/transforms.py index ecc9c54506..9767b629a7 100644 --- a/torchaudio/transforms.py +++ b/torchaudio/transforms.py @@ -611,7 +611,7 @@ def __init__(self, hop_length=None, n_freq=201, fixed_rate=None): n_fft = (n_freq - 1) * 2 hop_length = hop_length if hop_length is not None else n_fft // 2 self.register_buffer('phase_advance', torch.linspace(0, math.pi * hop_length, n_freq)[..., None]) - + def forward(self, complex_specgrams, overriding_rate=None): # type: (Tensor, Optional[float]) -> Tensor r"""