diff --git a/ppcls/arch/backbone/legendary_models/pp_hgnet.py b/ppcls/arch/backbone/legendary_models/pp_hgnet.py
index cc561ffe9..04a936f85 100644
--- a/ppcls/arch/backbone/legendary_models/pp_hgnet.py
+++ b/ppcls/arch/backbone/legendary_models/pp_hgnet.py
@@ -16,7 +16,8 @@ import paddle
 import paddle.nn as nn
 import paddle.nn.functional as F
 from paddle.nn.initializer import KaimingNormal, Constant
-from paddle.nn import Conv2D, BatchNorm2D, ReLU, AdaptiveAvgPool2D, MaxPool2D
+from paddle.nn import Conv2D, BatchNorm2D, ReLU, MaxPool2D
+from .custom_devices_layers import AdaptiveAvgPool2D
 from paddle.regularizer import L2Decay
 from paddle import ParamAttr