fix bug of resolution

fix bug of resolution

fix bug of resolution
fb0d3bf4 · cuicheng01 · GitHub · 8ad0665c · fb0d3bf4
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- 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,