From f49cf838a8acd23e17278cc60f3b52f6a13abe01 Mon Sep 17 00:00:00 2001
From: Jackwaterveg <87408988+Jackwaterveg@users.noreply.github.com>
Date: Tue, 25 Jan 2022 10:22:53 +0800
Subject: [PATCH] Update u2.py (#1378)

---
 paddlespeech/s2t/models/u2/u2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlespeech/s2t/models/u2/u2.py b/paddlespeech/s2t/models/u2/u2.py
index b6ec5f90..91079812 100644
--- a/paddlespeech/s2t/models/u2/u2.py
+++ b/paddlespeech/s2t/models/u2/u2.py
@@ -663,7 +663,7 @@ class U2BaseModel(ASRInterface, nn.Layer):
         # (num_hyps, max_hyps_len, vocab_size)
         decoder_out, _ = self.decoder(encoder_out, encoder_mask, hyps,
                                       hyps_lens)
-        decoder_out = paddle.nn.functional.log_softmax(decoder_out, dim=-1)
+        decoder_out = paddle.nn.functional.log_softmax(decoder_out, axis=-1)
         return decoder_out
 
     @paddle.no_grad()
-- 
GitLab