From 6370d519d2c58a97aa1495f5ee5277d6978b1f67 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Mon, 18 Oct 2021 17:23:08 +0800 Subject: [PATCH] Minor fixes. --- egs/librispeech/ASR/conformer_ctc_ali/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/conformer_ctc_ali/train.py b/egs/librispeech/ASR/conformer_ctc_ali/train.py index 2fb811dfb..391770bd0 100755 --- a/egs/librispeech/ASR/conformer_ctc_ali/train.py +++ b/egs/librispeech/ASR/conformer_ctc_ali/train.py @@ -379,10 +379,11 @@ def compute_loss( cut_ids=cut_ids, alignments=ali, num_classes=nnet_output.shape[2], + log_score=-1, ).to(nnet_output) min_len = min(nnet_output.shape[1], mask.shape[1]) - ali_scale = 500.0 / (params.batch_idx_train + 500) + ali_scale = 5000.0 / (params.batch_idx_train + 5000) nnet_output = nnet_output.clone() nnet_output[:, :min_len, :] += ali_scale * mask[:, :min_len, :]