Merge pull request #1926 from nemonameless/fix_ShuffleNetV2_x2_0_ch

fix ShuffleNetV2 config
2022-06-14 10:43:43 +08:00 · 2022-06-14 10:43:43 +08:00 · 61b7b3487a
parent ad068c60d1 60a4269d9b
commit 61b7b3487a
1 changed files with 1 additions and 1 deletions
--- a/ppcls/arch/backbone/model_zoo/shufflenet_v2.py
+++ b/ppcls/arch/backbone/model_zoo/shufflenet_v2.py
@ -233,7 +233,7 @@ class ShuffleNet(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!")