add attr min_row_size_to_use_multithread in op config test=develop

6 years ago · a6b3bf6069
parent 8c516a24e5
commit a6b3bf6069
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@ -822,7 +822,8 @@ class AdamOptimizer(Optimizer):
                "beta1": self._beta1,
                "beta2": self._beta2,
                "epsilon": self._epsilon,
-                "lazy_mode": self._lazy_mode
+                "lazy_mode": self._lazy_mode,
+                "min_row_size_to_use_multithread": 1000
            },
            stop_gradient=True)