未验证 提交 f6d485a8 编写于 作者: Z zmxdream 提交者: GitHub

fix trainer_pass.py (#37779)

上级 67cb5476
...@@ -771,7 +771,7 @@ def find_heter_ops(program, default_device="cpu"): ...@@ -771,7 +771,7 @@ def find_heter_ops(program, default_device="cpu"):
""" """
output_vars_no_grad = [] output_vars_no_grad = []
for key in pre_op.output_names: for key in op.output_names:
for varname in op.output(key): for varname in op.output(key):
if varname == "@EMPTY@": if varname == "@EMPTY@":
continue continue
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册