diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index f2a9cb53d..c7b910186 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index b28d8fb9a..ab81bbd4d 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index 4f65612ab..fc4de5464 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -590,31 +590,31 @@ def compute_loss( ) dense_fsa_vec_inter = [ - #k2.DenseFsaVec( - # nnet_output[1][2], - # supervision_segments, - # allow_truncate=params.subsampling_factor - 1, - #), - #k2.DenseFsaVec( - # nnet_output[1][5], - # supervision_segments, - # allow_truncate=params.subsampling_factor - 1, - #), + k2.DenseFsaVec( + nnet_output[1][2], + supervision_segments, + allow_truncate=params.subsampling_factor - 1, + ), + k2.DenseFsaVec( + nnet_output[1][5], + supervision_segments, + allow_truncate=params.subsampling_factor - 1, + ), k2.DenseFsaVec( nnet_output[1][8], supervision_segments, allow_truncate=params.subsampling_factor - 1, ), - #k2.DenseFsaVec( - # nnet_output[1][11], - # supervision_segments, - # allow_truncate=params.subsampling_factor - 1, - #), - #k2.DenseFsaVec( - # nnet_output[1][14], - # supervision_segments, - # allow_truncate=params.subsampling_factor - 1, - #) + k2.DenseFsaVec( + nnet_output[1][11], + supervision_segments, + allow_truncate=params.subsampling_factor - 1, + ), + k2.DenseFsaVec( + nnet_output[1][14], + supervision_segments, + allow_truncate=params.subsampling_factor - 1, + ) ] ctc_loss = k2.ctc_loss(