update

toolbox/torchaudio/models/spectrum_unet_irm/modeling_spectrum_unet_irm.py

@@ -394,8 +394,10 @@ class Encoder(nn.Module):
         # emb shape: [batch_size, time_steps, hidden_size * channels]
         emb, h = self.emb_gru.forward(emb, hidden_state)
 
-        lsnr = self.lsnr_fc(emb) * self.lsnr_scale + self.lsnr_offset
-
+        lsnr_ = self.lsnr_fc(emb)
+        print(f"lsnr_: {lsnr_}")
+        lsnr = lsnr_ * self.lsnr_scale + self.lsnr_offset
+        print(f"lsnr: {lsnr}")
         return e0, e1, e2, e3, emb, lsnr
 
 

toolbox/torchaudio/models/spectrum_unet_irm/yaml/config.yaml

@@ -26,8 +26,8 @@ encoder_emb_skip_op: "none"
 encoder_emb_linear_groups: 16
 encoder_emb_hidden_size: 256
 
-lsnr_max: 20
-lsnr_min: -10
+lsnr_max: 30
+lsnr_min: -15
 
 decoder_emb_num_layers: 3
 decoder_emb_skip_op: "none"