From 953c79caecf66760236767349401b2b595094da1 Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Tue, 27 Nov 2018 19:49:17 +0800 Subject: [PATCH] use dist_ctr to test remote_prefetch --- python/paddle/fluid/tests/unittests/dist_ctr.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/dist_ctr.py b/python/paddle/fluid/tests/unittests/dist_ctr.py index 902dc6544e..088f16a8ac 100644 --- a/python/paddle/fluid/tests/unittests/dist_ctr.py +++ b/python/paddle/fluid/tests/unittests/dist_ctr.py @@ -59,7 +59,8 @@ class TestDistCTR2x2(TestDistRunnerBase): param_attr=fluid.ParamAttr( name="deep_embedding", initializer=fluid.initializer.Constant(value=0.01)), - is_sparse=IS_SPARSE) + is_sparse=IS_SPARSE, + remote_prefetch=True) dnn_pool = fluid.layers.sequence_pool( input=dnn_embedding, pool_type="sum") dnn_out = dnn_pool @@ -81,7 +82,8 @@ class TestDistCTR2x2(TestDistRunnerBase): param_attr=fluid.ParamAttr( name="wide_embedding", 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") merge_layer = fluid.layers.concat(input=[dnn_out, lr_pool], axis=1) -- GitLab