From 8a788ad0d922c7d1b7c597a610b131f40c93e2b5 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 28 三月 2023 20:08:09 +0800
Subject: [PATCH] export
---
funasr/export/models/e2e_vad.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/export/models/e2e_vad.py b/funasr/export/models/e2e_vad.py
index 0653e06..b4236e0 100644
--- a/funasr/export/models/e2e_vad.py
+++ b/funasr/export/models/e2e_vad.py
@@ -11,7 +11,7 @@
class E2EVadModel(nn.Module):
def __init__(self, model,
max_seq_len=512,
- feats_dim=560,
+ feats_dim=400,
model_name='model',
**kwargs,):
super(E2EVadModel, self).__init__()
@@ -31,7 +31,7 @@
in_cache3: torch.Tensor,
):
- scores, cache0, cache1, cache2, cache3 = self.encoder(feats,
+ scores, (cache0, cache1, cache2, cache3) = self.encoder(feats,
in_cache0,
in_cache1,
in_cache2,
--
Gitblit v1.9.1