modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

志浩

2023-03-09 77cbfde968ae6a3d21677e16da2579611cd31737

modify unit test for speech_diarization_sond-en-us-callhome-8k-n16k4-pytorch

 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