From 6b0a4d6d4ed810d37db7d3b08695b6a8d642656c Mon Sep 17 00:00:00 2001 From: iamWHTWD <28421344+iamWHTWD@users.noreply.github.com> Date: Mon, 16 Jan 2023 14:15:00 +0800 Subject: [PATCH] Revert "recover transpose2's output for merging teacher model. (#1632)" This reverts commit a2a74a1921f5af2ec272e94d032da36c9b658955. --- paddleslim/dist/single_distiller.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/paddleslim/dist/single_distiller.py b/paddleslim/dist/single_distiller.py index 6d79ad67..8b20706a 100644 --- a/paddleslim/dist/single_distiller.py +++ b/paddleslim/dist/single_distiller.py @@ -14,7 +14,6 @@ import numpy as np import paddle -from ..common.recover_program import _recover_outputs_attr from paddleslim.core import GraphWrapper import paddle.nn.functional as F @@ -77,9 +76,7 @@ def merge(teacher_program, scope = paddle.static.global_scope() if teacher_scope == None: teacher_scope = scope - teacher_program = teacher_program.clone(for_test=True) - teacher_program = _recover_outputs_attr(teacher_program) is_same_model = True if len(student_program.blocks) == len(teacher_program.blocks): -- GitLab