fix bug of dense_table_set.add

767bf0c8 · xjqbest · dongdaxiang · 284adcc7 · 767bf0c8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/device_worker.py python/paddle/fluid/device_worker.py +1 -1

未找到文件。
--- a/python/paddle/fluid/device_worker.py
+++ b/python/paddle/fluid/device_worker.py
@@ -84,9 +84,9 @@ class DownpourSGD(DeviceWorker):
                    pc.push_sparse_table_id.extend([i])
                for i in program_configs[program_id]["push_dense"]:
                    pc.push_dense_table_id.extend([i])
+                    dense_table_set.add(i)
                for i in program_configs[program_id]["pull_sparse"]:
                    pc.pull_sparse_table_id.extend([i])
-                    dense_table_set.add(i)
                for i in program_configs[program_id]["pull_dense"]:
                    pc.pull_dense_table_id.extend([i])
                    dense_table_set.add(i)