From 1f85cf3aa48e719972165031ffa6fca6f7818994 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Sat, 25 Feb 2023 17:45:58 +0800 Subject: [PATCH] onnx --- funasr/export/models/e2e_asr_paraformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/export/models/e2e_asr_paraformer.py b/funasr/export/models/e2e_asr_paraformer.py index bf5ed1ea6..5424a0a94 100644 --- a/funasr/export/models/e2e_asr_paraformer.py +++ b/funasr/export/models/e2e_asr_paraformer.py @@ -59,7 +59,7 @@ class Paraformer(nn.Module): enc, enc_len = self.encoder(**batch) mask = self.make_pad_mask(enc_len)[:, None, :] pre_acoustic_embeds, pre_token_length, alphas, pre_peak_index = self.predictor(enc, mask) - pre_token_length = pre_token_length.round().type(torch.int32) + pre_token_length = pre_token_length.floor().type(torch.int32) decoder_out, _ = self.decoder(enc, enc_len, pre_acoustic_embeds, pre_token_length) decoder_out = torch.log_softmax(decoder_out, dim=-1)