update

toolbox/torchaudio/models/spectrum_unet_irm/modeling_spectrum_unet_irm.py

@@ -394,9 +394,9 @@ class Encoder(nn.Module):
         emb, h = self.emb_gru.forward(emb, hidden_state)
 
         lsnr_ = self.lsnr_fc(emb)
-        print(f"lsnr_: {lsnr_}")
+        print(f"lsnr_: {torch.any(torch.isnan(lsnr_))}")
         lsnr = lsnr_ * self.lsnr_scale + self.lsnr_offset
-        print(f"lsnr: {lsnr}")
+        print(f"lsnr: {torch.any(torch.isnan(lsnr))}")
         return e0, e1, e2, e3, emb, lsnr