mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
a973f6a903
commit
a2d8db09b1
Binary file not shown.
Binary file not shown.
@ -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)
|
||||
|
||||
|
||||
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user