From f2f3f6e758da73aa74c7db326b992249de35fd36 Mon Sep 17 00:00:00 2001 From: ziyoujiyi <73728031+ziyoujiyi@users.noreply.github.com> Date: Thu, 25 Aug 2022 17:27:12 +0800 Subject: [PATCH] Fl-PS bug fix (#45413) * back fl * delete ssl cert * . * make warning * . * unittest paral degree * solve unittest * heter & multi cloud commm ready * . * . * fl-ps v1.0 * . * support N + N mode * . * . * . * . * delete print * . * . * . * . * fix bug * . * . * fl-ps with coordinator ready * merge dev * update message parse only * update fl client scheduler * fix bug * update multithreads sync * fix ci errors * update role_maker.py * update role_maker.py * fix ci error: windows py import error * fix ci error: windows py import error * fix windows ci pylib import error * add dump fields & params * try to fix windows import fleet error * fix ps FLAGS error * fix logging risk * fix logging possible risk * write trainer_desc file * support split sparse params in local & remote * fix import paddle.fluid.core.PSGPU * fix import paddle.fluid.core.PSGPU * add remote_sparse & local_sparse config * fix unittest * fix test_dist_fleet_geo table error * fix PADDLE_ENFORCE error * fix other's pr conflict * forbidden ssd table * . * recover ssd table code * recover file mode --- python/paddle/distributed/ps/the_one_ps.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/paddle/distributed/ps/the_one_ps.py b/python/paddle/distributed/ps/the_one_ps.py index cebfda738d3..ea82f30cf8e 100755 --- a/python/paddle/distributed/ps/the_one_ps.py +++ b/python/paddle/distributed/ps/the_one_ps.py @@ -637,7 +637,6 @@ class SparseTable(Table): check_embedding_dim(table_proto.accessor, self.common.table_name, ctx.program_id(), self.context) - print(">>> set sparse table!") self.common.parse_by_optimizer(ctx, self.context) self.common.parse_entry(self.common.table_name, ctx.program_id(), self.context) @@ -806,6 +805,9 @@ class PsDescBuilder(object): not self.context['local_sparse']): tables.append(globals()['GeoSparseTable'](self.context, ctx)) + else: + tables.append(globals()['SparseTable'](self.context, + ctx)) else: tables.append(globals()['SparseTable'](self.context, ctx)) else: -- GitLab