From a12071ab8a99dfdb1aed4cb90f15090e0fdb54b6 Mon Sep 17 00:00:00 2001 From: hutuxian Date: Wed, 30 Sep 2020 13:52:11 +0800 Subject: [PATCH] add type check for initializer and regularizer in ParamAttr (#27720) --- python/paddle/fluid/param_attr.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/paddle/fluid/param_attr.py b/python/paddle/fluid/param_attr.py index 150168bc3ee..a76faf10590 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 == "": -- GitLab