export

游雁

2023-03-28 8a788ad0d922c7d1b7c597a610b131f40c93e2b5

 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,