From 07d6b123643f59377325f1611967f72505eaea8f Mon Sep 17 00:00:00 2001
From: yaozengwei <yaozengwei@outlook.com>
Date: Wed, 4 Sep 2024 19:33:41 +0800
Subject: [PATCH] update arguments of cr-ctc

---
 egs/librispeech/ASR/zipformer/train.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/egs/librispeech/ASR/zipformer/train.py b/egs/librispeech/ASR/zipformer/train.py
index 328b3cfdd..c2aaf8b56 100755
--- a/egs/librispeech/ASR/zipformer/train.py
+++ b/egs/librispeech/ASR/zipformer/train.py
@@ -460,10 +460,24 @@ def get_parser():
     parser.add_argument(
         "--cr-loss-scale",
         type=float,
-        default=0.1,
+        default=0.15,
         help="Scale for consistency-regularization loss.",
     )
 
+    parser.add_argument(
+        "--time-mask-ratio",
+        type=float,
+        default=2.0,
+        help="When using cr-ctc, we increase the time-masking ratio.",
+    )
+
+    parser.add_argument(
+        "--cr-loss-masked-scale",
+        type=float,
+        default=1.0,
+        help="The value used to scale up the cr_loss at masked positions",
+    )
+
     parser.add_argument(
         "--attention-decoder-loss-scale",
         type=float,
@@ -605,11 +619,6 @@ def get_params() -> AttributeDict:
             # parameters for attention-decoder
             "ignore_id": -1,
             "label_smoothing": 0.1,
-            # parameters used for CR-CTC
-            # When using cr-ctc, we increase the time-masking ratio.
-            "time_mask_ratio": 2.0,
-            # The scale used to scale up the cr_loss at masked positions.
-            "cr_loss_masked_scale": 3.0,
             "warm_step": 2000,
             "env_info": get_env_info(),
         }