diff --git a/ppdet/modeling/backbones/hrnet.py b/ppdet/modeling/backbones/hrnet.py index b74002cec82e5246c13db99be25e19098d45f001..6c4ef706f7eab19c48ada57d91a69ab9516d0651 100644 --- a/ppdet/modeling/backbones/hrnet.py +++ b/ppdet/modeling/backbones/hrnet.py @@ -15,6 +15,7 @@ import paddle import paddle.nn as nn import paddle.nn.functional as F +from paddle.nn import AdaptiveAvgPool2D, Linear, Uniform from paddle.regularizer import L2Decay from paddle import ParamAttr from paddle.nn.initializer import Normal