update

examples/conv_tasnet/run.sh

@@ -3,10 +3,10 @@
 : <<'END'
 
 
-sh run.sh --stage 1 --stop_stage 1 --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 200
+--max_epochs 400
 
 
 END

examples/conv_tasnet/step_2_train_model.py

@@ -47,6 +47,8 @@ def get_args():
 
     parser.add_argument("--config_file", default="config.yaml", type=str)
 
+    parser.add_argument("--seed", default=1234, type=int)
+
     args = parser.parse_args()
     return args
 
@@ -115,10 +117,10 @@ def main():
 
     logger = logging_config(serialization_dir)
 
-    random.seed(config.seed)
-    np.random.seed(config.seed)
-    torch.manual_seed(config.seed)
-    logger.info(f"set seed: {config.seed}")
+    random.seed(args.seed)
+    np.random.seed(args.seed)
+    torch.manual_seed(args.seed)
+    logger.info(f"set seed: {args.seed}")
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     n_gpu = torch.cuda.device_count()

requirements-python-3-9-9.txt

@@ -12,3 +12,4 @@ torch-pesq==0.1.2
 torchmetrics==1.6.1
 torchmetrics[audio]==1.6.1
 einops==0.8.1
+torch_stoi==0.2.3