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

@@ -149,7 +149,6 @@ def main():
         max_wave_value=32768.0,
         min_snr_db=config.min_snr_db,
         max_snr_db=config.max_snr_db,
-        # skip=625000,
     )
     train_data_loader = DataLoader(
         dataset=train_dataset,
@@ -254,6 +253,8 @@ def main():
     best_metric = None
     patience_count = 0
 
+    step_idx = 0 if last_step_idx == -1 else last_step_idx
+
     logger.info("training")
     for epoch_idx in range(max(0, last_epoch+1), args.max_epochs):
         # train
@@ -268,7 +269,6 @@ def main():
         total_pesq_loss = 0.
         total_batches = 0.
 
-        step_idx = 0 if last_step_idx == -1 else last_step_idx
         progress_bar_train = tqdm(
             initial=step_idx,
             desc="Training; epoch-{}".format(epoch_idx),

requirements.txt

@@ -1,4 +1,4 @@
-gradio==5.12.0
+gradio
 datasets==3.2.0
 python-dotenv==1.0.1
 scipy==1.15.1