fix r50vd bug of resolution

fix r50vd bug of resolution

fix r50vd bug of resolution
d74f4b44 · cuicheng01 · GitHub · 11d1c59c · d74f4b44
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppcls/modeling/architectures/resnet_vd.py ppcls/modeling/architectures/resnet_vd.py +1 -1

未找到文件。
--- a/ppcls/modeling/architectures/resnet_vd.py
+++ b/ppcls/modeling/architectures/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,