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-14 17:49:40 +09:00
parent d90ec45e33
commit e77b0077b3
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

5
egs/tedlium2/ASR/conformer_ctc3/transformer.py
View File

@ -249,7 +249,10 @@ class Transformer(nn.Module):
""" """
x = self.encoder_output_layer(x) x = self.encoder_output_layer(x)
x = x.permute(1, 0, 2) # (S, N, C) -> (N, S, C) x = x.permute(1, 0, 2) # (S, N, C) -> (N, S, C)
x = nn.functional.log_softmax(x, dim=-1) # (N, S, C) if log_prob:
x = nn.functional.log_softmax(x, dim=-1) # (N, S, C)
else:
x = nn.functional.softmax(x, dim=-1)
return x return x
@torch.jit.export @torch.jit.export