From b7bab1e6480cdea756682e2d5b831b0ba113b27f Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Thu, 23 Sep 2021 08:29:07 +0000 Subject: [PATCH] fix googlenet avg-pool --- ppcls/arch/backbone/model_zoo/googlenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/arch/backbone/model_zoo/googlenet.py b/ppcls/arch/backbone/model_zoo/googlenet.py index 00b7feeb..22528427 100644 --- 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( -- GitLab