From c7c97cb0fe91b06dcbc391b5a7dee92d496dc894 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyuthu@gmail.com>
Date: 星期四, 11 一月 2024 17:52:27 +0800
Subject: [PATCH] Update rwkv_encoder.py
---
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