fix srn for sublayer

3834ee10 · tink2123 · 2a529841 · 3834ee10
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

ppocr/modeling/heads/self_attention.py ppocr/modeling/heads/self_attention.py +1 -2

未找到文件。
--- a/ppocr/modeling/heads/self_attention.py
+++ b/ppocr/modeling/heads/self_attention.py
@@ -285,8 +285,7 @@ class PrePostProcessLayer(nn.Layer):
            elif cmd == "n":  # add layer normalization
                self.functors.append(
                    self.add_sublayer(
-                        "layer_norm_%d" % len(
+                        "layer_norm_%d" % len(self.sublayers()),
-                            self.sublayers(include_sublayers=False)),
                        paddle.nn.LayerNorm(
                            normalized_shape=d_model,
                            weight_attr=fluid.ParamAttr(