From fae856e23d45fd27d5fd55fd036e8e3fc7b24915 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期五, 02 六月 2023 23:00:08 +0800
Subject: [PATCH] update funasr-onnx-offline
---
funasr/modules/subsampling.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/modules/subsampling.py b/funasr/modules/subsampling.py
index 623be65..a2b91a7 100644
--- a/funasr/modules/subsampling.py
+++ b/funasr/modules/subsampling.py
@@ -506,9 +506,9 @@
)
self.conv = torch.nn.Sequential(
- torch.nn.Conv2d(1, conv_size, 3, 2),
+ torch.nn.Conv2d(1, conv_size, 3, 2, [1,0]),
torch.nn.ReLU(),
- torch.nn.Conv2d(conv_size, conv_size, kernel_2, stride_2),
+ torch.nn.Conv2d(conv_size, conv_size, kernel_2, stride_2, [(kernel_2-1)//2, 0]),
torch.nn.ReLU(),
)
@@ -597,7 +597,7 @@
mask: Mask of output sequences. (B, sub(T))
"""
if self.subsampling_factor > 1:
- return mask[:, :-2:2][:, : -(self.kernel_2 - 1) : self.stride_2]
+ return mask[:, ::2][:, ::self.stride_2]
else:
return mask
--
Gitblit v1.9.1