add frcrn model

examples/frcrn/run.sh

@@ -3,10 +3,9 @@
 : <<'END'
 
 
-sh run.sh --stage 2 --stop_stage 2 --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 frcrn \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise/dns3-noise" \
---speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech" \
---max_epochs 400
+--speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech"
 
 
 END

examples/frcrn/yaml/config.yaml

@@ -20,5 +20,8 @@ use_complex_networks: true
 model_depth: 20
 model_complexity: 45
 
+min_snr_db: -10
+max_snr_db: 20
+
 num_workers: 4
 batch_size: 4

toolbox/torchaudio/models/frcrn/configuration_frcrn.py

@@ -29,6 +29,9 @@ class FRCRNConfig(PretrainedConfig):
                  model_depth: int = 20,
                  model_complexity: int = 45,
 
+                 min_snr_db: float = -10,
+                 max_snr_db: float = 20,
+
                  num_workers: int = 4,
                  batch_size: int = 4,
                  **kwargs
@@ -53,6 +56,9 @@ class FRCRNConfig(PretrainedConfig):
         self.model_depth = model_depth
         self.model_complexity = model_complexity
 
+        self.min_snr_db = min_snr_db
+        self.max_snr_db = max_snr_db
+
         self.num_workers = num_workers
         self.batch_size = batch_size