diff --git a/paddleslim/nas/ofa/layers.py b/paddleslim/nas/ofa/layers.py
index f8edd9aab..1477b8070 100644
--- a/paddleslim/nas/ofa/layers.py
+++ b/paddleslim/nas/ofa/layers.py
@@ -20,7 +20,6 @@
 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_dygraph_mode
 
 __all__ = [
     'SuperConv2D', 'SuperConv2DTranspose', 'SuperSeparableConv2D',
@@ -985,7 +984,7 @@ def forward(self, input):
                  "use_global_stats", self._use_global_stats,
                  "trainable_statistics", trainable_statistics)
 
-        if in_dygraph_mode():
+        if paddle.in_dynamic_mode():
             paddle_compile = os.environ.get("paddle_compile")
             if feature_dim != self._mean.shape[0]:
                 if not paddle_compile or "Develop" in paddle_compile: