fix hrnet SELayer import (#3855)

1adca6bf · George Ni · GitHub · 6ea1ed6b · 1adca6bf
隐藏空白更改
内联并排

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