提交 a6b3bf60 编写于 作者: Q Qiao Longfei

add attr min_row_size_to_use_multithread in op config test=develop

上级 8c516a24
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册