diff --git a/ppcls/modeling/architectures/se_resnet_vd.py b/ppcls/modeling/architectures/se_resnet_vd.py index ef7dc3df2ab725e6996cb83eaa0663a54ae235dd..3267bb4c0498b0ba42cc3443cfae9a9a9a2d8342 100644 --- a/ppcls/modeling/architectures/se_resnet_vd.py +++ b/ppcls/modeling/architectures/se_resnet_vd.py @@ -45,7 +45,7 @@ class ConvBNLayer(fluid.dygraph.Layer): self.is_vd_mode = is_vd_mode self._pool2d_avg = Pool2D( - pool_size=2, pool_stride=2, pool_padding=0, pool_type='avg') + pool_size=2, pool_stride=2, pool_padding=0, pool_type='avg', ceil_mode=True) self._conv = Conv2D( num_channels=num_channels, num_filters=num_filters,