From 330447db0aa808c7a4a6c7a8c57ce4f9b34ad970 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Tue, 3 Jan 2023 10:02:40 +0800
Subject: [PATCH] adapt dev paddle framework api (#1616)

---
 paddleslim/nas/ofa/layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddleslim/nas/ofa/layers.py b/paddleslim/nas/ofa/layers.py
index bd2119ba..1162e921 100644
--- a/paddleslim/nas/ofa/layers.py
+++ b/paddleslim/nas/ofa/layers.py
@@ -20,7 +20,7 @@ import paddle
 from ...common import get_logger
 from .utils.utils import compute_start_end, get_same_padding, convert_to_list
 from .layers_base import *
-from paddle.framework import _in_legacy_dygraph, in_dygraph_mode
+from paddle.framework import in_dygraph_mode
 
 __all__ = [
     'SuperConv2D', 'SuperConv2DTranspose', 'SuperSeparableConv2D',
-- 
GitLab