update

examples/mpnet_aishell/run.sh

@@ -12,7 +12,7 @@ sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name fi
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --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 \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
 

examples/mpnet_aishell/step_2_train_model.py

@@ -44,8 +44,6 @@ def get_args():
 
     parser.add_argument("--max_epochs", default=100, type=int)
 
-    parser.add_argument("--batch_size", default=64, type=int)
-    parser.add_argument("--learning_rate", default=1e-4, type=float)
     parser.add_argument("--num_serialized_models_to_keep", default=10, type=int)
     parser.add_argument("--patience", default=5, type=int)
     parser.add_argument("--serialization_dir", default="serialization_dir", type=str)
@@ -142,7 +140,7 @@ def main():
     )
     train_data_loader = DataLoader(
         dataset=train_dataset,
-        batch_size=args.batch_size,
+        batch_size=config.batch_size,
         shuffle=True,
         sampler=None,
         # Linux 系统中可以使用多个子进程加载数据, 而在 Windows 系统中不能.
@@ -153,7 +151,7 @@ def main():
     )
     valid_data_loader = DataLoader(
         dataset=valid_dataset,
-        batch_size=args.batch_size,
+        batch_size=config.batch_size,
         shuffle=True,
         sampler=None,
         # Linux 系统中可以使用多个子进程加载数据, 而在 Windows 系统中不能.

requirements-python-3-9-9.txt

@@ -8,7 +8,7 @@ openpyxl==3.1.5
 torch==2.5.1
 torchaudio==2.5.1
 overrides==7.7.0
-torch-pesq
-torchmetrics
-torchmetrics[audio]
-einops
+torch-pesq==0.1.2
+torchmetrics==1.6.1
+torchmetrics[audio]==1.6.1
+einops==0.8.1