未验证 提交 bbb8b366 编写于 作者: Z zhoushiyu 提交者: GitHub

deepfm dygraph update optimizer (#4141)

上级 f26154fc
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册