From 759425adbf9193a2e64bdec89a05f87c0c810925 Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Mon, 12 Nov 2018 22:38:39 +0800 Subject: [PATCH] add is_sparse and is_distributed --- fluid/PaddleRec/ctr/network_conf.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/fluid/PaddleRec/ctr/network_conf.py b/fluid/PaddleRec/ctr/network_conf.py index 269d83aa..c1eb1bed 100644 --- a/fluid/PaddleRec/ctr/network_conf.py +++ b/fluid/PaddleRec/ctr/network_conf.py @@ -15,8 +15,10 @@ def ctr_dnn_model(embedding_size, sparse_feature_dim): def embedding_layer(input): return fluid.layers.embedding( input=input, + is_sparse=True, + is_distributed=True, size=[sparse_feature_dim, embedding_size], - param_attr=fluid.ParamAttr(name="SparseFeatFactors", initializer=fluid.initializer.Normal(scale=1/math.sqrt(sparse_feature_dim)))) + param_attr=fluid.ParamAttr(name="SparseFeatFactors", initializer=fluid.initializer.fluid.initializer.Uniform())) sparse_embed_seq = map(embedding_layer, sparse_input_ids) concated = fluid.layers.concat(sparse_embed_seq + [dense_input], axis=1) -- GitLab