From c8bae0ec85eee25d66de6b1e4502eff74d750b24 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 21 十二月 2023 13:29:37 +0800
Subject: [PATCH] funasr2
---
funasr/models/fsmn_vad/encoder.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/funasr/models/fsmn_vad/fsmn_encoder.py b/funasr/models/fsmn_vad/encoder.py
similarity index 98%
rename from funasr/models/fsmn_vad/fsmn_encoder.py
rename to funasr/models/fsmn_vad/encoder.py
index 38d164d..50e31fc 100755
--- a/funasr/models/fsmn_vad/fsmn_encoder.py
+++ b/funasr/models/fsmn_vad/encoder.py
@@ -6,6 +6,8 @@
import torch.nn as nn
import torch.nn.functional as F
+from funasr.utils.register import register_class, registry_tables
+
class LinearTransform(nn.Module):
def __init__(self, input_dim, output_dim):
@@ -156,7 +158,7 @@
fsmn_layers: no. of sequential fsmn layers
'''
-
+@register_class("encoder_classes", "FSMN")
class FSMN(nn.Module):
def __init__(
self,
@@ -227,7 +229,7 @@
rstride: right stride
'''
-
+@register_class("encoder_classes", "DFSMN")
class DFSMN(nn.Module):
def __init__(self, dimproj=64, dimlinear=128, lorder=20, rorder=1, lstride=1, rstride=1):
--
Gitblit v1.9.1