提交 953c79ca 编写于 作者: Q Qiao Longfei

use dist_ctr to test remote_prefetch

上级 18fd2d01
...@@ -59,7 +59,8 @@ class TestDistCTR2x2(TestDistRunnerBase): ...@@ -59,7 +59,8 @@ class TestDistCTR2x2(TestDistRunnerBase):
param_attr=fluid.ParamAttr( param_attr=fluid.ParamAttr(
name="deep_embedding", name="deep_embedding",
initializer=fluid.initializer.Constant(value=0.01)), initializer=fluid.initializer.Constant(value=0.01)),
is_sparse=IS_SPARSE) is_sparse=IS_SPARSE,
remote_prefetch=True)
dnn_pool = fluid.layers.sequence_pool( dnn_pool = fluid.layers.sequence_pool(
input=dnn_embedding, pool_type="sum") input=dnn_embedding, pool_type="sum")
dnn_out = dnn_pool dnn_out = dnn_pool
...@@ -81,7 +82,8 @@ class TestDistCTR2x2(TestDistRunnerBase): ...@@ -81,7 +82,8 @@ class TestDistCTR2x2(TestDistRunnerBase):
param_attr=fluid.ParamAttr( param_attr=fluid.ParamAttr(
name="wide_embedding", name="wide_embedding",
initializer=fluid.initializer.Constant(value=0.01)), initializer=fluid.initializer.Constant(value=0.01)),
is_sparse=IS_SPARSE) is_sparse=IS_SPARSE,
remote_prefetch=True)
lr_pool = fluid.layers.sequence_pool(input=lr_embbding, pool_type="sum") lr_pool = fluid.layers.sequence_pool(input=lr_embbding, pool_type="sum")
merge_layer = fluid.layers.concat(input=[dnn_out, lr_pool], axis=1) merge_layer = fluid.layers.concat(input=[dnn_out, lr_pool], axis=1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册