From 369f2ff8838569351c26d909407b497ab16d3bcf Mon Sep 17 00:00:00 2001
From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com>
Date: 星期二, 20 六月 2023 11:08:50 +0800
Subject: [PATCH] add html5 local access asr service directly (#654)
---
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