未验证 提交 3e8b3220 编写于 作者: Z ziyoujiyi 提交者: GitHub

cherry-pick V2.4 (#46294)

* back fl

* delete ssl cert

* .

* make warning

* .

* unittest paral degree

* solve unittest

* heter & multi cloud commm ready

* .

* .

* fix gloo compile warning

* adapt for nn fl-ps

* flps del fake-init op

* add learning_rate_0 intializer op
上级 461099c0
......@@ -617,6 +617,8 @@ class DeleteExtraOptimizerPass(PassBase):
for var in remote_optimize_vars:
if var in local_optimize_vars:
continue
if 'learning_rate_0' == var:
continue
if var not in remote_optimize_op_role_vars:
optimize_need_delete_vars.append(var)
need_delete_optimize_vars = list(set(optimize_need_delete_vars))
......
......@@ -373,8 +373,8 @@ class FlPsProgramBuilder(HeterAsyncPsProgramBuilder):
_main_file = ps_log_root_dir + '4_fl_worker_main_program.prototxt'
#debug_program(_main_file, self.cloned_main)
fake_init_ops_pass = new_pass("fake_init_ops_pass", self.attrs)
fake_init_ops_pass.apply([None], [self.cloned_startup], self.pass_ctx)
#fake_init_ops_pass = new_pass("fake_init_ops_pass", self.attrs)
#fake_init_ops_pass.apply([None], [self.cloned_startup], self.pass_ctx)
_main_file = ps_log_root_dir + '5_fl_worker_main_program.prototxt'
#debug_program(_main_file, self.cloned_main)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册