提交 008e0c9b 编写于 作者: X Xin Pan

small clean

上级 52993878
...@@ -301,6 +301,7 @@ class DistributeTranspiler(object): ...@@ -301,6 +301,7 @@ class DistributeTranspiler(object):
Program: trainer side program. Program: trainer side program.
""" """
# remove optimize ops and add a send op to main_program # remove optimize ops and add a send op to main_program
# FIXME(typhoonzero): Also ops like clip_gradient, lrn_decay?
delete_ops(self.origin_program.global_block(), self.optimize_ops) delete_ops(self.origin_program.global_block(), self.optimize_ops)
self.origin_program.__str__() self.origin_program.__str__()
return self.origin_program return self.origin_program
...@@ -537,7 +538,6 @@ class DistributeTranspiler(object): ...@@ -537,7 +538,6 @@ class DistributeTranspiler(object):
# 2. rename op outputs # 2. rename op outputs
for op in orig_s_prog.global_block().ops: for op in orig_s_prog.global_block().ops:
new_inputs = dict()
new_outputs = dict() new_outputs = dict()
# do not append startup op if var is not on this pserver # do not append startup op if var is not on this pserver
op_on_pserver = False op_on_pserver = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册