diff --git a/dygraph/models/hrnet.py b/dygraph/models/hrnet.py
index bccc303bb435e48554991a21b4fd72dd90a3cb37..d5270a1089c46d390a05ca5c53df2d8e044a900c 100644
--- a/dygraph/models/hrnet.py
+++ b/dygraph/models/hrnet.py
@@ -19,7 +19,10 @@ import paddle.fluid as fluid
 from paddle.fluid.param_attr import ParamAttr
 from paddle.fluid.layer_helper import LayerHelper
 from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear
-from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
+try:
+    from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
+except:
+    from paddle.fluid.dygraph import BatchNorm
 
 __all__ = [
     "HRNet_W18_Small_V1", "HRNet_W18_Small_V2", "HRNet_W18", "HRNet_W30",
diff --git a/dygraph/models/unet.py b/dygraph/models/unet.py
index 6e04c8b2f17aeca763dc9653b6b2da73835979c7..9bb92302f5a65eac6f536f0308f01ea878361aa7 100644
--- a/dygraph/models/unet.py
+++ b/dygraph/models/unet.py
@@ -14,7 +14,10 @@
 
 import paddle.fluid as fluid
 from paddle.fluid.dygraph import Conv2D, Pool2D
-from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
+try:
+    from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
+except:
+    from paddle.fluid.dygraph import BatchNorm
 
 
 class UNet(fluid.dygraph.Layer):