adapt AdaptiveAvgPool2D for npu for PPHGNet (#3162)

2024-06-17 21:07:16 +08:00 · 2024-06-17 21:07:16 +08:00 · cea80378ac
parent be6e0b7b53
commit cea80378ac
1 changed files with 2 additions and 1 deletions
--- 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