瀏覽代碼

update repo

嘉渊 2 年之前
父節點
當前提交
f3fb62cd4f
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      funasr/models/e2e_vad.py

+ 2 - 1
funasr/models/e2e_vad.py

@@ -5,6 +5,7 @@ import torch
 from torch import nn
 import math
 from funasr.models.encoder.fsmn_encoder import FSMN
+from funasr.models.base_model import FunASRModel
 
 
 class VadStateMachine(Enum):
@@ -211,7 +212,7 @@ class WindowDetector(object):
         return int(self.frame_size_ms)
 
 
-class E2EVadModel(nn.Module):
+class E2EVadModel(FunASRModel):
     """
     Author: Speech Lab of DAMO Academy, Alibaba Group
     Deep-FSMN for Large Vocabulary Continuous Speech Recognition