Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 10 days ago

Commit

a88007c

1 Parent(s): 35a4689

update

Files changed (1) hide show

examples/dfnet/step_2_train_model.py CHANGED Viewed

@@ -264,7 +264,7 @@ def main():
             clean_audios: torch.Tensor = clean_audios.to(device)
             noisy_audios: torch.Tensor = noisy_audios.to(device)
-            est_spec, est_wav, est_mask, lsnr, erb_encoder_h = model.forward(noisy_audios)
             mr_stft_loss = mr_stft_loss_fn.forward(est_wav, clean_audios)
             neg_si_snr_loss = neg_si_snr_loss_fn.forward(est_wav, clean_audios)
@@ -335,7 +335,7 @@ def main():
                         clean_audios: torch.Tensor = clean_audios.to(device)
                         noisy_audios: torch.Tensor = noisy_audios.to(device)
-                        est_spec, est_wav, est_mask, lsnr, erb_encoder_h = model.forward(noisy_audios)
                         mr_stft_loss = mr_stft_loss_fn.forward(est_wav, clean_audios)
                         neg_si_snr_loss = neg_si_snr_loss_fn.forward(est_wav, clean_audios)

             clean_audios: torch.Tensor = clean_audios.to(device)
             noisy_audios: torch.Tensor = noisy_audios.to(device)
+            est_spec, est_wav, est_mask, lsnr = model.forward(noisy_audios)
             mr_stft_loss = mr_stft_loss_fn.forward(est_wav, clean_audios)
             neg_si_snr_loss = neg_si_snr_loss_fn.forward(est_wav, clean_audios)
                         clean_audios: torch.Tensor = clean_audios.to(device)
                         noisy_audios: torch.Tensor = noisy_audios.to(device)
+                        est_spec, est_wav, est_mask, lsnr = model.forward(noisy_audios)
                         mr_stft_loss = mr_stft_loss_fn.forward(est_wav, clean_audios)
                         neg_si_snr_loss = neg_si_snr_loss_fn.forward(est_wav, clean_audios)