LayerList add sublayer none check (#22135)

test=develop

LayerList add sublayer none check (#22135)
test=develop
d0406193 · songyouwei · hong · 0c2fff44 · d0406193
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

python/paddle/fluid/dygraph/container.py python/paddle/fluid/dygraph/container.py +3 -2

未找到文件。
--- a/python/paddle/fluid/dygraph/container.py
+++ b/python/paddle/fluid/dygraph/container.py
@@ -201,8 +201,9 @@ class LayerList(Layer):
    def __init__(self, sublayers=None):
        super(LayerList, self).__init__()
-        for idx, layer in enumerate(sublayers):
+        if sublayers is not None:
-            self.add_sublayer(str(idx), layer)
+            for idx, layer in enumerate(sublayers):
+                self.add_sublayer(str(idx), layer)
    def __getitem__(self, idx):
        if isinstance(idx, slice):