diff --git a/python/paddle/fluid/param_attr.py b/python/paddle/fluid/param_attr.py index 150168bc3ee4e89de3fcacaf1a2150923ac056d0..a76faf10590684e09f361cef3c6bbd5f3b9ce799 100644 --- 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 == "":