fix

b1fdf253 · chengmo · 9a51e107 · b1fdf253
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

core/engine/cluster/cluster.py core/engine/cluster/cluster.py +4 -2

未找到文件。
--- a/core/engine/cluster/cluster.py
+++ b/core/engine/cluster/cluster.py
@@ -170,13 +170,15 @@ class ClusterEnvBase(object):
                "config.communicator.FLAGS_communicator_is_sgd_optimizer", 0)
        self.cluster_env[
            "FLAGS_communicator_send_queue_size"] = self.backend_env.get(
-                "config.communicator.FLAGS_communicator_send_queue_size", max_thread_num)
+                "config.communicator.FLAGS_communicator_send_queue_size",
+                max_thread_num)
        self.cluster_env[
            "FLAGS_communicator_thread_pool_size"] = self.backend_env.get(
                "config.communicator.FLAGS_communicator_thread_pool_size", 32)
        self.cluster_env[
            "FLAGS_communicator_max_merge_var_num"] = self.backend_env.get(
-                "config.communicator.FLAGS_communicator_max_merge_var_num", max_thread_num)
+                "config.communicator.FLAGS_communicator_max_merge_var_num",
+                max_thread_num)
        self.cluster_env[
            "FLAGS_communicator_max_send_grad_num_before_recv"] = self.backend_env.get(
                "config.communicator.FLAGS_communicator_max_send_grad_num_before_recv",