adapt AdaptiveAvgPool2D for npu for PPHGNet (#3163) (#3223)

2024-08-26 11:57:09 +08:00 · 2024-08-26 11:57:09 +08:00 · 3dac3be506
parent 82034d1b3f
commit 3dac3be506
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