提交 0a9249c9 编写于 作者: C chenguowei01

update hrnet.py

上级 38f472e4
...@@ -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.
先完成此消息的编辑!
想要评论请 注册