fix hrnet SELayer import (#3856)

e3c993c5 · George Ni · GitHub · 2a2644d6 · e3c993c5
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

ppdet/modeling/backbones/hrnet.py ppdet/modeling/backbones/hrnet.py +1 -0

未找到文件。
--- 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