diff --git a/dygraph/models/hrnet.py b/dygraph/models/hrnet.py index f97d93cf300123a85fca8f4d56c223b8a778534e..3c8139d14a9d9afa2c3679210fe579db3c579502 100644 --- a/dygraph/models/hrnet.py +++ b/dygraph/models/hrnet.py @@ -22,8 +22,10 @@ from paddle.fluid.dygraph.nn import Conv2D, Pool2D, Linear from paddle.fluid.initializer import Normal try: from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm + print('using sync batch norm') except: from paddle.fluid.dygraph import BatchNorm + print('using batch norm') __all__ = [ "HRNet_W18_Small_V1", "HRNet_W18_Small_V2", "HRNet_W18", "HRNet_W30",