update

examples/frcrn/run.sh

@@ -3,7 +3,7 @@
 : <<'END'
 
 
-sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name frcrn \
+sh run.sh --stage 2 --stop_stage 2 --system_version centos --file_folder_name file_dir --final_model_name frcrn-20-512-nx-dns3 \
 --config_file "yaml/config-20-512.yaml" \
 --noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
 --speech_dir "/data/tianxing/HuggingDatasets/nx_noise/data/speech/dns3-speech"

examples/frcrn/step_2_train_model.py

@@ -41,11 +41,11 @@ from toolbox.torchaudio.models.frcrn.modeling_frcrn import FRCRN, FRCRNPretraine
 
 def get_args():
     parser = argparse.ArgumentParser()
-    parser.add_argument("--train_dataset", default="train.xlsx", type=str)
-    parser.add_argument("--valid_dataset", default="valid.xlsx", type=str)
+    parser.add_argument("--train_dataset", default="train.jsonl", type=str)
+    parser.add_argument("--valid_dataset", default="valid.jsonl", type=str)
 
     parser.add_argument("--num_serialized_models_to_keep", default=15, type=int)
-    parser.add_argument("--patience", default=10, type=int)
+    parser.add_argument("--patience", default=30, type=int)
     parser.add_argument("--serialization_dir", default="serialization_dir", type=str)
 
     parser.add_argument("--config_file", default="config.yaml", type=str)
@@ -185,7 +185,7 @@ def main():
         step_idx = int(step_idx)
         if step_idx > last_step_idx:
             last_step_idx = step_idx
-    last_epoch = 0
+    # last_epoch = 0
 
     if last_step_idx != -1:
         logger.info(f"resume from steps-{last_step_idx}.")