diff --git a/dygraph/models/hrnet.py b/dygraph/models/hrnet.py index fac8a929be40acce2d801c3cdbbe89bb634bead3..bccc303bb435e48554991a21b4fd72dd90a3cb37 100644 --- a/dygraph/models/hrnet.py +++ b/dygraph/models/hrnet.py @@ -18,7 +18,8 @@ import paddle import paddle.fluid as fluid from paddle.fluid.param_attr import ParamAttr from paddle.fluid.layer_helper import LayerHelper -from paddle.fluid.dygraph.nn import Conv2D, Pool2D, BatchNorm, Linear +from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear +from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm __all__ = [ "HRNet_W18_Small_V1", "HRNet_W18_Small_V2", "HRNet_W18", "HRNet_W30",