Skip to content

P
PaddleSeg

PaddlePaddle / PaddleSeg

通知 285
Star 8
Fork 1
P
PaddleSeg
提交 b7f59d54 编写于 作者: R root
浏览文件
操作

add sync_bn

上级 06d51479
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
dygraph/models/unet.py
浏览文件 @ b7f59d54
...@@ -13,7 +13,8 @@ ...@@ -13,7 +13,8 @@
# limitations under the License. # limitations under the License.
import paddle.fluid as fluid 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): class UNet(fluid.dygraph.Layer):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或