Skip to content

D
DeepSpeech

PaddlePaddle / DeepSpeech
大约 2 年 前同步成功

通知 210
Star 8425
Fork 1598
D
DeepSpeech
未验证 提交 f49cf838 编写于 作者: J Jackwaterveg 提交者： GitHub
浏览文件
操作

Update u2.py (#1378)

上级 b26c1efe
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
paddlespeech/s2t/models/u2/u2.py
浏览文件 @ f49cf838
...@@ -663,7 +663,7 @@ class U2BaseModel(ASRInterface, nn.Layer): ...@@ -663,7 +663,7 @@ class U2BaseModel(ASRInterface, nn.Layer):
# (num_hyps, max_hyps_len, vocab_size) # (num_hyps, max_hyps_len, vocab_size)
decoder_out, _ = self.decoder(encoder_out, encoder_mask, hyps, decoder_out, _ = self.decoder(encoder_out, encoder_mask, hyps,
hyps_lens) 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 return decoder_out
@paddle.no_grad() @paddle.no_grad()
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或