The error first comes In “loudness_extractor.py”, which containers the first part of the encoder,  so I set a pdb.set_trace() in there:
   ```

“”"
def forward(self, z):
Compute A-weighted Loudness Extraction
Input:
z[‘audio’] : batch of time-domain signals
Output:
output_signal : batch of reverberated signals
“”"
    input_signal = z['audio']
    paded_input_signal = nn.functional.pad(input_signal, (self.frame_length * 2, self.frame_length * 2))
    sliced_signal = paded_input_signal.unfold(1, self.n_fft, self.frame_length)
    pdb.set_trace()
    sliced_windowed_signal = sliced_signal * self.smoothing_window

I set the pdb.set_trace() and did self.smoothing_window.cuda and got: device(type='cuda', index=0) and then i also did sliced_signal.get_device() and got 0.