Update u2.py (#1378)

3 years ago · f49cf838a8
parent b26c1efe0f
commit f49cf838a8
1 changed files with 1 additions and 1 deletions
--- 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()