update

examples/clean_unet_aishell/test.py

@@ -1,12 +0,0 @@
-#!/usr/bin/python3
-# -*- coding: utf-8 -*-
-
-
-nn.Sequential(
-    nn.Conv1d(channels_h, channels_h * 2, 1),
-    nn.GLU(dim=1),
-    nn.ConvTranspose1d(channels_h, channels_output, kernel_size, stride),
-    # nn.ReLU()
-)
-if __name__ == '__main__':
-    pass

toolbox/torchaudio/models/clean_unet/loss.py

@@ -141,6 +141,7 @@ class MultiResolutionSTFTLoss(torch.nn.Module):
         mag_loss = 0.0
         for f in self.stft_losses:
             sc_l, mag_l = f(x, y)
+            print(f"sc_l: {sc_l}, mag_l: {mag_l}")
             sc_loss += sc_l
             mag_loss += mag_l