Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 10 days ago

Commit

c255825

1 Parent(s): a88007c

update

Files changed (1) hide show

toolbox/torchaudio/models/dfnet/modeling_dfnet.py CHANGED Viewed

@@ -982,8 +982,8 @@ class DfNet(nn.Module):
             raise AssertionError("Input signals must have the same shape")
         noise = noisy - clean
-        clean, _ = self.signal_prepare(clean)
-        noise, _ = self.signal_prepare(noise)
         stft_clean = self.stft.forward(clean)
         mag_clean = torch.abs(stft_clean)
@@ -1002,8 +1002,8 @@ class DfNet(nn.Module):
             raise AssertionError("Input signals must have the same shape")
         noise = noisy - clean
-        clean, _ = self.signal_prepare(clean)
-        noise, _ = self.signal_prepare(noise)
         stft_clean = self.stft.forward(clean)
         stft_noise = self.stft.forward(noise)

             raise AssertionError("Input signals must have the same shape")
         noise = noisy - clean
+        clean = self.signal_prepare(clean)
+        noise = self.signal_prepare(noise)
         stft_clean = self.stft.forward(clean)
         mag_clean = torch.abs(stft_clean)
             raise AssertionError("Input signals must have the same shape")
         noise = noisy - clean
+        clean = self.signal_prepare(clean)
+        noise = self.signal_prepare(noise)
         stft_clean = self.stft.forward(clean)
         stft_noise = self.stft.forward(noise)