paraformer onnx fp16导出方案 (#2264)

will_wang

2024-12-04 0c3c9be2c4c1c4e4da4628c3987708c9a0763391

 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