Merge pull request #2692 from tink2123/fix_srn_for_sublayers

Fix srn for sublayers

Merge pull request #2692 from tink2123/fix_srn_for_sublayers
Fix srn for sublayers
0f3c5917 · MissPenguin · GitHub · 0ccc1720 · 1720e0c8 · 0f3c5917
隐藏空白更改
内联并排

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(