HoneyTian commited on
Commit
2abaf72
·
1 Parent(s): 54e8f54
examples/clean_unet_aishell/test.py DELETED
@@ -1,12 +0,0 @@
1
- #!/usr/bin/python3
2
- # -*- coding: utf-8 -*-
3
-
4
-
5
- nn.Sequential(
6
- nn.Conv1d(channels_h, channels_h * 2, 1),
7
- nn.GLU(dim=1),
8
- nn.ConvTranspose1d(channels_h, channels_output, kernel_size, stride),
9
- # nn.ReLU()
10
- )
11
- if __name__ == '__main__':
12
- pass
 
 
 
 
 
 
 
 
 
 
 
 
 
toolbox/torchaudio/models/clean_unet/loss.py CHANGED
@@ -141,6 +141,7 @@ class MultiResolutionSTFTLoss(torch.nn.Module):
141
  mag_loss = 0.0
142
  for f in self.stft_losses:
143
  sc_l, mag_l = f(x, y)
 
144
  sc_loss += sc_l
145
  mag_loss += mag_l
146
 
 
141
  mag_loss = 0.0
142
  for f in self.stft_losses:
143
  sc_l, mag_l = f(x, y)
144
+ print(f"sc_l: {sc_l}, mag_l: {mag_l}")
145
  sc_loss += sc_l
146
  mag_loss += mag_l
147