funasr/models/whisper_lid/eres2net/simple_avg.py
@@ -3,6 +3,7 @@ from funasr.models.encoder.abs_encoder import AbsEncoder from funasr.modules.nets_utils import make_pad_mask class SimpleAvg(AbsEncoder): def __init__(self, feat_dim): super(SimpleAvg, self).__init__() @@ -14,4 +15,4 @@ return avg_x def output_size(self) -> int: return self.feat_dim return self.feat_dim