嘉渊 hace 2 años
padre
commit
f3fb62cd4f
Se han modificado 1 ficheros con 2 adiciones y 1 borrados
  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