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

… to be created is larger than 0.
PaddlePaddle · Jun 1, 2021 · 25df091 · 25df091 · paddle-bot-old · Jun 2, 2021
1 parent 0192b82
commit 25df091
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