From 84d8b3ae034a4a366001779c41dfaf4ebe196e77 Mon Sep 17 00:00:00 2001
From: root <root@yq01-qianmo-com-255-129-12.yq01.baidu.com>
Date: Mon, 6 Jul 2020 14:19:39 +0000
Subject: [PATCH] add sync_bn

---
 dygraph/models/hrnet.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dygraph/models/hrnet.py b/dygraph/models/hrnet.py
index fac8a929..bccc303b 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",
-- 
GitLab