use dist_ctr to test remote_prefetch

f7c96f079b
Qiao Longfei 6 years ago
parent 18fd2d01b7
commit 953c79caec

@ -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):
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)

Loading…
Cancel
Save