Skip to content

P
PaddleSeg

PaddlePaddle / PaddleSeg

通知 285
Star 8
Fork 1
P
PaddleSeg
提交 0a9249c9 编写于 作者: C chenguowei01
浏览文件
操作

update hrnet.py

上级 38f472e4
显示空白变更内容
内联 并排
Showing with 2 addition and 0 deletion
dygraph/models/hrnet.py
浏览文件 @ 0a9249c9
...@@ -22,8 +22,10 @@ from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear ...@@ -22,8 +22,10 @@ from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear
from paddle.fluid.initializer import Normal from paddle.fluid.initializer import Normal
try: try:
from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
print('using sync batch norm')
except: except:
from paddle.fluid.dygraph import BatchNorm from paddle.fluid.dygraph import BatchNorm
print('using batch norm')
__all__ = [ __all__ = [
"HRNet_W18_Small_V1", "HRNet_W18_Small_V2", "HRNet_W18", "HRNet_W30", "HRNet_W18_Small_V1", "HRNet_W18_Small_V2", "HRNet_W18", "HRNet_W30",
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或