add sync_bn

b7f59d54 · root · 06d51479 · b7f59d54
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

dygraph/models/unet.py dygraph/models/unet.py +2 -1

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