From 8606c9a55d0eaebc1fec9fc67b59238ad38da134 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Fri, 17 Feb 2023 10:47:03 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc2/.train.py.swp | Bin 65536 -> 65536 bytes egs/librispeech/ASR/conformer_ctc2/train.py | 20 ++++-------------- 2 files changed, 4 insertions(+), 16 deletions(-) diff --git a/egs/librispeech/ASR/conformer_ctc2/.train.py.swp b/egs/librispeech/ASR/conformer_ctc2/.train.py.swp index 9e8e4b1da1c73023c99a24f30f28590036d8ed6f..a53d063f994ba1baeccf74718f21f91ffb8662ca 100644 GIT binary patch delta 308 zcmZo@U}@>q$REfx1!7Ske#ywdu$po5fi~I63mjP| zZ*X*)Ea1a3`P>psAfFA$7npo-iPYqYjv|w5T{tIedht#^mu)#Y)kA6X6(<8;z2kfg z4F7=`WIQ{N0)gZF3=DICI3I}ffY=F$Re|_F-)6&ry?m1oH1bS7*TxUE&~S2nydERN z1d3mWN@%g1C1*Ij?2C)nZNFXy!p`@}PRRc+hrh=`lLP=>sPO8;pK_`pN J|Kgbz005WBR`dV> delta 239 zcmZo@U}+{& diff --git a/egs/librispeech/ASR/conformer_ctc2/train.py b/egs/librispeech/ASR/conformer_ctc2/train.py index 2c2ea6192..2534be386 100755 --- a/egs/librispeech/ASR/conformer_ctc2/train.py +++ b/egs/librispeech/ASR/conformer_ctc2/train.py @@ -677,22 +677,10 @@ def compute_loss( ctc_loss = (1-params.interctc_weight) * ctc_loss + params.interctc_weight * inter_ctc_loss - if not params.interctc and not params.condition and params.group_num > 0: - dense_fsa_vec = k2.DenseFsaVec( - nnet_output[0], - supervision_segments, - allow_truncate=params.subsampling_factor - 1, - ) - - ctc_loss = k2.ctc_loss( - decoding_graph=decoding_graph, - dense_fsa_vec=dense_fsa_vec, - output_beam=params.beam_size, - reduction=params.reduction, - use_double_scores=params.use_double_scores, - ) - - if not params.interctc and not params.condition and params.group_num == 0: + if not params.interctc and not params.condition: + if type(nnet_output) == tuple: + nnet_output = nnet_output[0] + dense_fsa_vec = k2.DenseFsaVec( nnet_output, supervision_segments,