提交 6b0a4d6d 编写于 作者: I iamWHTWD 提交者: GitHub

Revert "recover transpose2's output for merging teacher model. (#1632)"

This reverts commit a2a74a19.
上级 785004f5
...@@ -14,7 +14,6 @@ ...@@ -14,7 +14,6 @@
import numpy as np import numpy as np
import paddle import paddle
from ..common.recover_program import _recover_outputs_attr
from paddleslim.core import GraphWrapper from paddleslim.core import GraphWrapper
import paddle.nn.functional as F import paddle.nn.functional as F
...@@ -77,9 +76,7 @@ def merge(teacher_program, ...@@ -77,9 +76,7 @@ def merge(teacher_program,
scope = paddle.static.global_scope() scope = paddle.static.global_scope()
if teacher_scope == None: if teacher_scope == None:
teacher_scope = scope teacher_scope = scope
teacher_program = teacher_program.clone(for_test=True) teacher_program = teacher_program.clone(for_test=True)
teacher_program = _recover_outputs_attr(teacher_program)
is_same_model = True is_same_model = True
if len(student_program.blocks) == len(teacher_program.blocks): if len(student_program.blocks) == len(teacher_program.blocks):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册