update

examples/mpnet_aishell/run.sh

@@ -17,6 +17,11 @@ sh run.sh --stage 5 --stop_stage 5 --system_version centos --file_folder_name fi
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 
 
+sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \
+--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
+--speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech"
+
+
 END
 
 

main.py

@@ -88,8 +88,11 @@ def main():
     # engines
     global denoise_engines
     denoise_engines = {
-        "mpnet-aishell-20250224": InferenceMPNet(
-            pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet-aishell-20250224.zip").as_posix(),
+        "mpnet-aishell-1-epoch": InferenceMPNet(
+            pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet-aishell-1-epoch.zip").as_posix(),
+        ),
+        "mpnet-aishell-11-epoch": InferenceMPNet(
+            pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet-aishell-11-epoch.zip").as_posix(),
         ),
     }