From e77b0077b38dfa46c87a6fd662f68f1e1fbb47c6 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 14 Feb 2023 17:49:40 +0900
Subject: [PATCH] from local

---
 .../ASR/conformer_ctc3/.transformer.py.swp    | Bin 57344 -> 61440 bytes
 .../ASR/conformer_ctc3/transformer.py         |   5 ++++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp
index cf35417bc7add7d92dc8dc9061b376191cfa56ec..e392a0ead2f5036f9dfc137ce7835ba05fc9396e 100644
GIT binary patch
delta 644
zcmY+?Pe_w-9LMqReD+V<<Nh2TL4}?}hXpAsEHOwy52hmu!aR+!Ws_CUR@1|bU_0qS
zgz9k!mB{Q6blQ5zb*VhmWd&t)Xt%;s6b3<gP{j944}Rgr^ZPx|^WpjZzP~5Gs^q^Z
zT0C@R=z`wU84}7TvOWCss<BbletB)GsrU?A(7f$y@i*5WQ{DeBvSUs8#=XyVdr7Uj
zMW!3MPdaG&KRwCSXZI1tFuKqN6~B3V6^}843`Wq4Q#ghIzPLon$RGkgK0C?cDGIoY
z>uASEhsZLD4jGku49-JC6TaFB;Sw&Q9X}M2d6<|)7&?yPLxaeB+(8K6Z6XhB*6i9p
zSF3)gk_r80dOUh7la8OY9@Y}hTx?9_6$^U7g6l0g=-6woNC`QlU|<xz2qA!V+WZF3
zP(T(jj9?I_v8mD=6p%+3{(8uw>XDqhWUz!e^x@ADk#~3t3pdb%4s10M!UQttg9jB(
zY6S~OVF3N8pWX%*aT#{(ZFK*);0vm~;xy9{xAkbP>M~<BEijQzOd7FlFndBDOB&}o
OP98Ul(Khp1?aVF=xp3nE

delta 309
zcmXBPy-PxI6vpv$y}lHhUJDe$3pb+CP^7`BSu)VY8qA@WDH=kWBywz(pSp-kE(Q+f
z{RfKHf(z1cX_1Bk4eU?w8y@(;8J_3BsZaF!vmQH5m2!txb~h!AA!2UN&XTWBvEPUm
zQo3=amDRNoxBJF%q~Ta2%R5{PkW|_-URF`3cXzTXlE(q|kU$uJD<UuG;vQ9$u#XIO
zuz?Rwj!;Df>lpj#qK`IiaS0QSPo#$jG<<R-IhHtraJ`gJL;)tIx=0I6+#ruFSa|n{
eyrG6PzIg5by+_R+{Coz~xtmbm^G&<sUicS`eK@-S

diff --git a/egs/tedlium2/ASR/conformer_ctc3/transformer.py b/egs/tedlium2/ASR/conformer_ctc3/transformer.py
index 17878ca90..4ff9062fe 100644
--- a/egs/tedlium2/ASR/conformer_ctc3/transformer.py
+++ b/egs/tedlium2/ASR/conformer_ctc3/transformer.py
@@ -249,7 +249,10 @@ class Transformer(nn.Module):
         """
         x = self.encoder_output_layer(x)
         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
 
     @torch.jit.export