from local

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

View File

@ -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)

View File

@ -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,