mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-18 05:32:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

update arguments of cr-ctc

Browse Source
This commit is contained in:
yaozengwei 2024-09-04 19:33:41 +08:00
parent ebbbcbcaf1
commit 07d6b12364
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
egs/librispeech/ASR/zipformer/train.py
View File

@ -460,10 +460,24 @@ def get_parser():
parser.add_argument( parser.add_argument(
"--cr-loss-scale", "--cr-loss-scale",
type=float, type=float,
default=0.1, default=0.15,
help="Scale for consistency-regularization loss.", 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( parser.add_argument(
"--attention-decoder-loss-scale", "--attention-decoder-loss-scale",
type=float, type=float,
@ -605,11 +619,6 @@ def get_params() -> AttributeDict:
# parameters for attention-decoder # parameters for attention-decoder
"ignore_id": -1, "ignore_id": -1,
"label_smoothing": 0.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, "warm_step": 2000,
"env_info": get_env_info(), "env_info": get_env_info(),
} }