From a5cd4bb473b19fe2af1753fa7e60a997a8208447 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期一, 14 八月 2023 10:54:23 +0800
Subject: [PATCH] support offline inference for unified streaming/non-streaming rnnt
---
funasr/modules/subsampling.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/modules/subsampling.py b/funasr/modules/subsampling.py
index 77aa422..af33aef 100644
--- a/funasr/modules/subsampling.py
+++ b/funasr/modules/subsampling.py
@@ -358,7 +358,8 @@
"""
x = x.transpose(1, 2) # (b, d ,t)
x = self.pad_fn(x)
- x = F.relu(self.conv(x))
+ #x = F.relu(self.conv(x))
+ x = F.leaky_relu(self.conv(x), negative_slope=0.)
x = x.transpose(1, 2) # (b, t ,d)
if x_len is None:
--
Gitblit v1.9.1