Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 18 days ago

Commit

0598200

1 Parent(s): 9d91461

update

Files changed (1) hide show

examples/conv_tasnet/step_2_train_model.py CHANGED Viewed

@@ -324,6 +324,7 @@ def main():
                     total_ae_loss = 0.
                     total_neg_si_snr_loss = 0.
                     total_neg_stoi_loss = 0.
                     total_batches = 0.
                     progress_bar_train.close()
@@ -384,6 +385,7 @@ def main():
                     total_ae_loss = 0.
                     total_neg_si_snr_loss = 0.
                     total_neg_stoi_loss = 0.
                     total_batches = 0.
                     progress_bar_eval.close()

                     total_ae_loss = 0.
                     total_neg_si_snr_loss = 0.
                     total_neg_stoi_loss = 0.
+                    total_mr_stft_loss = 0.
                     total_batches = 0.
                     progress_bar_train.close()
                     total_ae_loss = 0.
                     total_neg_si_snr_loss = 0.
                     total_neg_stoi_loss = 0.
+                    total_mr_stft_loss = 0.
                     total_batches = 0.
                     progress_bar_eval.close()