update

examples/nx_clean_unet/run.sh

@@ -12,9 +12,10 @@ sh run.sh --stage 3 --stop_stage 3 --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 5 --stop_stage 5 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
+sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
---speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
+--speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train" \
+--max_epochs 100 --max_count 10000
 
 
 sh run.sh --stage 1 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name mpnet-nx-speech-20250224 \

examples/nx_clean_unet/yaml/config.yaml

@@ -21,3 +21,5 @@ discriminator_dim: 32
 discriminator_in_channel: 2
 
 compress_factor: 0.3
+
+seed: 1234

toolbox/torchaudio/models/nx_clean_unet/configuration_nx_clean_unet.py

@@ -28,6 +28,7 @@ class NXCleanUNetConfig(PretrainedConfig):
 
                  compress_factor: float = 0.3,
 
+                 seed: int = 1234,
                  **kwargs
                  ):
         super(NXCleanUNetConfig, self).__init__(**kwargs)
@@ -51,6 +52,8 @@ class NXCleanUNetConfig(PretrainedConfig):
 
         self.compress_factor = compress_factor
 
+        self.seed = seed
+
 
 if __name__ == '__main__':
     pass

toolbox/torchaudio/models/nx_clean_unet/yaml/config.yaml

@@ -21,3 +21,5 @@ discriminator_dim: 32
 discriminator_in_channel: 2
 
 compress_factor: 0.3
+
+seed: 1234