From 718d61c578b533674b339c20551d503318820697 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Wed, 5 Apr 2017 11:22:04 +0800
Subject: [PATCH] Disable sparse_remote_update when local in paddle.v2

---
 python/paddle/v2/trainer.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/paddle/v2/trainer.py b/python/paddle/v2/trainer.py
index 7bd3e2c565..4e432a52b2 100644
--- a/python/paddle/v2/trainer.py
+++ b/python/paddle/v2/trainer.py
@@ -52,6 +52,12 @@ class SGD(object):
         self.__topology__ = topology
         self.__parameters__ = parameters
         self.__topology_in_proto__ = topology.proto()
+
+        # In local mode, disable sparse_remote_update.
+        for param in self.__topology_in_proto__.parameters:
+            if param.sparse_remote_update:
+                param.sparse_remote_update = False
+
         self.__data_types__ = topology.data_type()
         gm = api.GradientMachine.createFromConfigProto(
             self.__topology_in_proto__, api.CREATE_MODE_NORMAL,
-- 
GitLab