aky15
2023-03-15 e33bb15d269bb3e2e41f7a3540d9b92703bb5c50
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