update

examples/conv_tasnet/run.sh

@@ -3,7 +3,7 @@
 : <<'END'
 
 
-sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name conv-tasnet-dns3-20250319 \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name conv-tasnet-dns3-20250319 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise/dns3-noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech" \
 --max_epochs 400

examples/conv_tasnet/step_2_train_model.py

@@ -238,6 +238,7 @@ def main():
         total_neg_stoi_loss = 0.
         total_mr_stft_loss = 0.
         total_batches = 0.
+
         total_steps = 0
         progress_bar_train = tqdm(
             desc="Training; epoch-{}".format(idx_epoch),
@@ -356,6 +357,14 @@ def main():
                             "neg_stoi_loss": average_neg_stoi_loss,
                             "mr_stft_loss": average_mr_stft_loss,
                         })
+
+                    total_pesq_score = 0.
+                    total_loss = 0.
+                    total_ae_loss = 0.
+                    total_neg_si_snr_loss = 0.
+                    total_neg_stoi_loss = 0.
+                    total_batches = 0.
+
                     progress_bar_eval.close()
                     progress_bar_train = tqdm(
                         initial=progress_bar_train.n,