diff --git a/python/paddle/distributed/ps/the_one_ps.py b/python/paddle/distributed/ps/the_one_ps.py index cebfda738d3025dcaa41db7ac93caf76c99923aa..ea82f30cf8ec027ed0b55adee4b4f838c7fcfbd0 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: