add an assertion to ensure that the size of each dim of the parameter…

… to be created is larger than 0. (#33265)
PaddlePaddle · Jun 3, 2021 · 23b9ed3 · 23b9ed3
1 parent 3bbf2d7
commit 23b9ed3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/fluid/layer_helper_base.py b/python/paddle/fluid/layer_helper_base.py
@@ -312,6 +312,10 @@ def create_parameter(self,
         if not attr:
             return None
         assert isinstance(attr, ParamAttr)
+        for i, size in enumerate(shape):
+            assert size > 0, (
+                "Expected every dim's size to be larger than 0, "
+                "but the size of the {}-th dim is {}".format(i, size))
         # set global dtype
         if not dtype:
             dtype = self.__dtype