未验证 提交 5f00305d 编写于 作者: W WangZhen 提交者: GitHub

Fix load fine tune error (#55233)

上级 32bc8b88
...@@ -338,6 +338,7 @@ class _ProgramHolder: ...@@ -338,6 +338,7 @@ class _ProgramHolder:
self._output_descs = [] self._output_descs = []
self._double_grad_descs = [] self._double_grad_descs = []
self._persistable_names = [] self._persistable_names = []
self._grad_var_names = {}
# execution scope # execution scope
self._inner_scope = core.Scope() self._inner_scope = core.Scope()
...@@ -350,7 +351,6 @@ class _ProgramHolder: ...@@ -350,7 +351,6 @@ class _ProgramHolder:
self._train_program_desc = self._append_backward_desc( self._train_program_desc = self._append_backward_desc(
self._infer_program_desc self._infer_program_desc
) )
self._grad_var_names = {}
# forward: # forward:
@switch_to_static_graph @switch_to_static_graph
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册