mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-15 14:50:30 +09:00
parent a973f6a903
commit a2d8db09b1
4 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp
View File

Binary file not shown.

BIN
egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
View File

Binary file not shown.

2
egs/tedlium2/ASR/conformer_ctc3/conformer.py
View File

@ -454,6 +454,8 @@ class ConformerEncoder(nn.Module):
outputs.append(output) outputs.append(output)
if i+1 in [3,6,9,12,15] and condition_layer is not None: 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) ctc_out = ctc_output(output, log_prob=False)
output = output + condition_layer(ctc_out).transpose(0,1) output = output + condition_layer(ctc_out).transpose(0,1)

1
egs/tedlium2/ASR/conformer_ctc3/train.py
View File

@ -663,6 +663,7 @@ def compute_loss(
reduction=params.reduction, reduction=params.reduction,
use_double_scores=params.use_double_scores, use_double_scores=params.use_double_scores,
) )
else: else:
dense_fsa_vec = k2.DenseFsaVec( dense_fsa_vec = k2.DenseFsaVec(
nnet_output, nnet_output,