From 909cfbb477b82211b8ecd3def11eaf7efa5e20d3 Mon Sep 17 00:00:00 2001 From: Feng Ni Date: Tue, 28 Jun 2022 17:46:02 +0800 Subject: [PATCH] fix ShuffleNetV2 x2_0 stage_out_channels (#6139) --- ppdet/modeling/backbones/shufflenet_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/backbones/shufflenet_v2.py b/ppdet/modeling/backbones/shufflenet_v2.py index 996697ad7..09bff0602 100644 --- a/ppdet/modeling/backbones/shufflenet_v2.py +++ b/ppdet/modeling/backbones/shufflenet_v2.py @@ -188,7 +188,7 @@ class ShuffleNetV2(nn.Layer): elif scale == 1.5: stage_out_channels = [-1, 24, 176, 352, 704, 1024] elif scale == 2.0: - stage_out_channels = [-1, 24, 224, 488, 976, 2048] + stage_out_channels = [-1, 24, 244, 488, 976, 2048] else: raise NotImplementedError("This scale size:[" + str(scale) + "] is not implemented!") -- GitLab