zhaomingwork
2023-05-19 b63e73ae4f5df9d4ed9fb0bee12ac2cc09d7f523
funasr/export/models/modules/encoder_layer.py
@@ -61,7 +61,7 @@
        if self.feed_forward_macaron is not None:
            residual = x
            x = self.norm_ff_macaron(x)
            x = residual + self.feed_forward_macaron(x)
            x = residual + self.feed_forward_macaron(x) * 0.5
        residual = x
        x = self.norm_mha(x)
@@ -81,7 +81,7 @@
        residual = x
        x = self.norm_ff(x)
        x = residual + self.feed_forward(x)
        x = residual + self.feed_forward(x) * 0.5
        x = self.norm_final(x)