HoneyTian commited on
Commit
bd5c606
·
1 Parent(s): 20acb96
Files changed (2) hide show
  1. examples/mpnet_aishell/run.sh +1 -1
  2. main.py +2 -2
examples/mpnet_aishell/run.sh CHANGED
@@ -12,7 +12,7 @@ sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name fi
12
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
13
  --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
14
 
15
- sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
16
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
17
  --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
18
 
 
12
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
13
  --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
14
 
15
+ sh run.sh --stage 5 --stop_stage 5 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
16
  --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
17
  --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
18
 
main.py CHANGED
@@ -88,8 +88,8 @@ def main():
88
  # engines
89
  global denoise_engines
90
  denoise_engines = {
91
- "mpnet_aishell_20250221": InferenceMPNet(
92
- pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet_aishell_20250221.zip").as_posix(),
93
  ),
94
  }
95
 
 
88
  # engines
89
  global denoise_engines
90
  denoise_engines = {
91
+ "mpnet_aishell_20250224": InferenceMPNet(
92
+ pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet_aishell_20250224.zip").as_posix(),
93
  ),
94
  }
95