From 07bd9f3e0ea0adf105dd7e49a7d1d3dbeb1f5ddc Mon Sep 17 00:00:00 2001 From: Dongji Gao Date: Tue, 19 Sep 2023 14:22:46 -0400 Subject: [PATCH] add --blank-bias in decoding.py --- egs/librispeech/WSASR/conformer_ctc2/decode.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/WSASR/conformer_ctc2/decode.py b/egs/librispeech/WSASR/conformer_ctc2/decode.py index 8f6f1847f..32906716a 100755 --- a/egs/librispeech/WSASR/conformer_ctc2/decode.py +++ b/egs/librispeech/WSASR/conformer_ctc2/decode.py @@ -68,7 +68,14 @@ def get_parser(): ) parser.add_argument( - "--otc-token", type=str, default="", help="OTC token", + "--otc-token", type=str, default="▁", help="OTC token", + ) + + parser.add_argument( + "--blank-bias", + type=float, + default=0, + help="bias (log-prob) added to blank token during decoding", ) parser.add_argument( @@ -384,6 +391,7 @@ def decode_one_batch( nnet_output, memory, memory_key_padding_mask = model(feature, supervisions) # nnet_output is (N, T, C) + nnet_output[:, :, 0] += params.blank_bias supervision_segments = torch.stack( (