From b7f59d5484516470e0bea56ea17025ebf3fac1e7 Mon Sep 17 00:00:00 2001
From: root <root@yq01-qianmo-com-255-129-12.yq01.baidu.com>
Date: Tue, 7 Jul 2020 12:21:30 +0000
Subject: [PATCH] add sync_bn

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

diff --git a/dygraph/models/unet.py b/dygraph/models/unet.py
index b55e3614..6e04c8b2 100644
--- a/dygraph/models/unet.py
+++ b/dygraph/models/unet.py
@@ -13,7 +13,8 @@
 # limitations under the License.
 
 import paddle.fluid as fluid
-from paddle.fluid.dygraph import Conv2D, BatchNorm, Pool2D
+from paddle.fluid.dygraph import Conv2D, Pool2D
+from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
 
 
 class UNet(fluid.dygraph.Layer):
-- 
GitLab