提交 5bfdec61 编写于 作者: Q Qiao Longfei

set is_distributed to false

上级 1e9aaf2d
......@@ -16,7 +16,9 @@ def ctr_dnn_model(embedding_size, sparse_feature_dim):
return fluid.layers.embedding(
input=input,
is_sparse=True,
is_distributed=True,
# you need to patch https://github.com/PaddlePaddle/Paddle/pull/14190
# if you want to set is_distributed to True
is_distributed=False,
size=[sparse_feature_dim, embedding_size],
param_attr=fluid.ParamAttr(name="SparseFeatFactors", initializer=fluid.initializer.Uniform()))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册