From 3698ca040f2be2572e64d6bec9753fe3b27021b0 Mon Sep 17 00:00:00 2001 From: pkufool Date: Thu, 21 Nov 2024 12:10:48 +0800 Subject: [PATCH] Minor fixes --- egs/libriheavy/ASR/zipformer/train.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/egs/libriheavy/ASR/zipformer/train.py b/egs/libriheavy/ASR/zipformer/train.py index 0a566e2ca4..bf2ee7db9e 100644 --- a/egs/libriheavy/ASR/zipformer/train.py +++ b/egs/libriheavy/ASR/zipformer/train.py @@ -471,13 +471,6 @@ def get_parser(): 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, @@ -966,9 +959,8 @@ def compute_loss( spec_augment=spec_augment, supervision_segments=supervision_segments, time_warp_factor=params.spec_aug_time_warp_factor, - cr_loss_masked_scale=params.cr_loss_masked_scale, ) - simple_loss, pruned_loss, ctc_loss = losses[:3] + simple_loss, pruned_loss, ctc_loss, attention_decoder_loss, cr_loss = losses[:5] loss = 0.0