add type check for initializer and regularizer in ParamAttr (#27720)

4 years ago · a12071ab8a
parent fd616fadc2
commit a12071ab8a
1 changed files with 4 additions and 0 deletions
--- a/python/paddle/fluid/param_attr.py
+++ b/python/paddle/fluid/param_attr.py
@ -87,6 +87,10 @@ class ParamAttr(object):
        check_type(learning_rate, "learning_rate", (float, int), "ParamAttr")
        check_type(trainable, "trainable", (bool), "ParamAttr")
        check_type(do_model_average, "do_model_average", (bool), "ParamAttr")
+        check_type(initializer, "initializer", (Initializer, type(None)),
+                   "ParamAttr")
+        check_type(regularizer, "regularizer",
+                   (WeightDecayRegularizer, type(None)), "ParamAttr")

        self.name = name
        if self.name == "":