diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index 23d9fb8e8..103e13057 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 614c37d3e..a03356c02 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/conformer.py b/egs/tedlium2/ASR/conformer_ctc3/conformer.py index e33e683d0..9ddd20f49 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/conformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/conformer.py @@ -454,6 +454,8 @@ class ConformerEncoder(nn.Module): outputs.append(output) if i+1 in [3,6,9,12,15] and condition_layer is not None: + exit() + #TODO ctc_out = ctc_output(output, log_prob=False) output = output + condition_layer(ctc_out).transpose(0,1) diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index 9bc3fdbfa..c04e7e690 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -663,6 +663,7 @@ def compute_loss( reduction=params.reduction, use_double_scores=params.use_double_scores, ) + else: dense_fsa_vec = k2.DenseFsaVec( nnet_output,