train

游雁

2023-05-16 3d9f094e9652d4b84894c6fd4eae39a4a753b0f0

 funasr/models/frontend/fused.py

@@ -1,3 +1,4 @@
from funasr.models.frontend.abs_frontend import AbsFrontend
from funasr.models.frontend.default import DefaultFrontend
from funasr.models.frontend.s3prl import S3prlFrontend
import numpy as np
@@ -6,7 +7,7 @@
from typing import Tuple


class FusedFrontends(torch.nn.Module):
class FusedFrontends(AbsFrontend):
    def __init__(
        self, frontends=None, align_method="linear_projection", proj_dim=100, fs=16000
    ):
@@ -142,4 +143,4 @@
        else:
            raise NotImplementedError

        return input_feats, feats_lens
        return input_feats, feats_lens