From 90d92111cf317b56509a96c7fb88d1a0bcfb7897 Mon Sep 17 00:00:00 2001
From: wanghuancoder <wanghuancoder@163.com>
Date: Fri, 5 Feb 2021 17:04:10 +0800
Subject: [PATCH] let LayerList could add [None], test=develop (#30911)

---
 python/paddle/fluid/dygraph/layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/fluid/dygraph/layers.py b/python/paddle/fluid/dygraph/layers.py
index b9a43cbbe1..b157ce81d8 100644
--- a/python/paddle/fluid/dygraph/layers.py
+++ b/python/paddle/fluid/dygraph/layers.py
@@ -959,7 +959,7 @@ class Layer(core.Layer):
                 for prefix, layer in model.named_sublayers():
                     print(prefix, layer)
         """
-        assert isinstance(sublayer, core.Layer)
+        assert (isinstance(sublayer, core.Layer) or sublayer == None)
 
         self._sub_layers[name] = sublayer
         return sublayer
-- 
GitLab