Module GPU test fixes (#369)

* Fixed GPU tests

Author: jaeyeun97

torchaudio/transforms.py

@@ -51,7 +51,7 @@ def __init__(self, n_fft=400, win_length=None, hop_length=None,
         self.win_length = win_length if win_length is not None else n_fft
         self.hop_length = hop_length if hop_length is not None else self.win_length // 2
         window = window_fn(self.win_length) if wkwargs is None else window_fn(self.win_length, **wkwargs)
-        self.window = window
+        self.register_buffer('window', window)
         self.pad = pad
         self.power = power
         self.normalized = normalized
@@ -136,7 +136,7 @@ def __init__(self, n_mels=128, sample_rate=16000, f_min=0., f_max=None, n_stft=N
 
         fb = torch.empty(0) if n_stft is None else F.create_fb_matrix(
             n_stft, self.f_min, self.f_max, self.n_mels, self.sample_rate)
-        self.fb = fb
+        self.register_buffer('fb', fb)
 
     def forward(self, specgram):
         r"""
@@ -260,7 +260,7 @@ def __init__(self, sample_rate=16000, n_mfcc=40, dct_type=2, norm='ortho', log_m
         if self.n_mfcc > self.MelSpectrogram.n_mels:
             raise ValueError('Cannot select more MFCC coefficients than # mel bins')
         dct_mat = F.create_dct(self.n_mfcc, self.MelSpectrogram.n_mels, self.norm)
-        self.dct_mat = dct_mat
+        self.register_buffer('dct_mat', dct_mat)
         self.log_mels = log_mels
 
     def forward(self, waveform):