From 72b6c7735e44fff12c4a3d401deb98c59f16ab0f Mon Sep 17 00:00:00 2001
From: hnluo <haoneng.lhn@alibaba-inc.com>
Date: 星期二, 29 八月 2023 11:40:46 +0800
Subject: [PATCH] fix onnx export bug
---
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