志浩
2023-03-09 77cbfde968ae6a3d21677e16da2579611cd31737
funasr/models/encoder/opennmt_encoders/conv_encoder.py
@@ -137,12 +137,12 @@
            self.out_padding = nn.ConstantPad1d((left_padding, right_padding), 0.0)
            self.conv_out = nn.Conv1d(
                num_units,
                num_units,
                out_units,
                kernel_size,
        )
        if self.out_norm:
            self.after_norm = LayerNorm(num_units)
            self.after_norm = LayerNorm(out_units)
    def output_size(self) -> int:
        return self.num_units