fix googlenet avg-pool

2021-09-23 08:29:07 +00:00 · 2021-09-23 08:29:07 +00:00 · b7bab1e648
parent 9a4b05cfc6
commit b7bab1e648
1 changed files with 1 additions and 1 deletions
--- a/ppcls/arch/backbone/model_zoo/googlenet.py
+++ b/ppcls/arch/backbone/model_zoo/googlenet.py
@ -131,7 +131,7 @@ class GoogLeNetDY(nn.Layer):
        self._ince5b = Inception(
            832, 832, 384, 192, 384, 48, 128, 128, name="ince5b")

-        self._pool_5 = AvgPool2D(kernel_size=7, stride=7)
+        self._pool_5 = AdaptiveAvgPool2D(1)

        self._drop = Dropout(p=0.4, mode="downscale_in_infer")
        self._fc_out = Linear(