diff --git a/PaddleRec/ctr/deepfm_dygraph/train.py b/PaddleRec/ctr/deepfm_dygraph/train.py index 19774e971342b87f052ae4edecb947f9109d7f81..97a7626cc325fac708e37913229c51ea09acbd44 100644 --- a/PaddleRec/ctr/deepfm_dygraph/train.py +++ b/PaddleRec/ctr/deepfm_dygraph/train.py @@ -66,6 +66,7 @@ def train(args): (epoch, auc_metric_test.eval())) optimizer = fluid.optimizer.Adam( + parameter_list=deepfm.parameters(), regularization=fluid.regularizer.L2DecayRegularizer(args.reg)) # load model if exists