Skip to content

Commit cd81075

Browse files
authored
Merge pull request #1745 from reyoung/feature/fix_sparse_local_update
Disable sparse_remote_update when local in paddle.v2
2 parents 93007fe + 718d61c commit cd81075

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

python/paddle/v2/trainer.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,12 @@ def __init__(self, cost, parameters, update_equation):
5252
self.__topology__ = topology
5353
self.__parameters__ = parameters
5454
self.__topology_in_proto__ = topology.proto()
55+
56+
# In local mode, disable sparse_remote_update.
57+
for param in self.__topology_in_proto__.parameters:
58+
if param.sparse_remote_update:
59+
param.sparse_remote_update = False
60+
5561
self.__data_types__ = topology.data_type()
5662
gm = api.GradientMachine.createFromConfigProto(
5763
self.__topology_in_proto__, api.CREATE_MODE_NORMAL,

0 commit comments

Comments
 (0)