diff --git a/paddlespeech/s2t/models/u2/u2.py b/paddlespeech/s2t/models/u2/u2.py index b6ec5f90d84ed639a300d1456a6c0fce0ba01ee9..910798127ee5c8c7c00893b603ec3ef95dc5be26 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()