From 8c24f52fd25fce2df46dc4d9dffb45619dc38f9f Mon Sep 17 00:00:00 2001
From: aky15 <ankeyuthu@gmail.com>
Date: 星期三, 24 五月 2023 23:40:03 +0800
Subject: [PATCH] Dev aky2 (#549)
---
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